Katpeeler commited on
Commit
1f4a6b2
·
1 Parent(s): 7d3afbb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -117,6 +117,7 @@ def process(num1, num2, num3):
117
  BAR_LENGTH_120BPM = 4.0 * 60 / num3
118
  input_ids = tokenizer.encode(created_text, return_tensors="pt")
119
  generated_ids = model.generate(input_ids, max_length=500)
 
120
  generated_sequence = tokenizer.decode(generated_ids[0])
121
 
122
  # Convert text of notes to audio
@@ -128,13 +129,9 @@ def process(num1, num2, num3):
128
  array_of_floats *= 32767
129
  int16_data = array_of_floats.astype(np.int16)
130
  return SAMPLE_RATE, int16_data
131
- return generated_sequence
132
 
133
 
134
  def generation(text):
135
- input_ids = tokenizer.encode(text, return_tensors="pt")
136
- generated_ids = model.generate(input_ids, max_length=500)
137
- generated_sequence = tokenizer.decode(generated_ids[0])
138
  return generated_sequence
139
 
140
 
@@ -156,8 +153,8 @@ with gr.Blocks() as demo:
156
  number3 = gr.Slider(60, 140, value=120, label="BPM", step=5, info="Choose between 60 and 140")
157
  audio_button = gr.Button("generate audio")
158
 
159
- text_button.click(generation, inputs=text_input, outputs=text_output)
160
- audio_button.click(process, inputs=[number1, number2, number3], outputs=[audio_output, text_output2])
161
 
162
 
163
  if __name__ == "__main__":
 
117
  BAR_LENGTH_120BPM = 4.0 * 60 / num3
118
  input_ids = tokenizer.encode(created_text, return_tensors="pt")
119
  generated_ids = model.generate(input_ids, max_length=500)
120
+ global generated_sequence
121
  generated_sequence = tokenizer.decode(generated_ids[0])
122
 
123
  # Convert text of notes to audio
 
129
  array_of_floats *= 32767
130
  int16_data = array_of_floats.astype(np.int16)
131
  return SAMPLE_RATE, int16_data
 
132
 
133
 
134
  def generation(text):
 
 
 
135
  return generated_sequence
136
 
137
 
 
153
  number3 = gr.Slider(60, 140, value=120, label="BPM", step=5, info="Choose between 60 and 140")
154
  audio_button = gr.Button("generate audio")
155
 
156
+ text_button.click(generation, outputs=text_output)
157
+ audio_button.click(process, inputs=[number1, number2, number3], outputs=audio_output)
158
 
159
 
160
  if __name__ == "__main__":