Update app.py

app.py

@@ -25,20 +25,26 @@ def get_note_text(prompt):
     )
     return response.choices[0].text.strip()
 
-def get_drummer_output(prompt):
+def get_drummer_output(prompt, tempo):
     openai.api_key = os.environ['key']
+    if tempo == "fast":
+        tempo = 138
+    elif tempo == "slow":
+        tempo = 100
     note_text = get_note_text(prompt)
     # note_text = note_text + " " + note_text
-    prompt_enc = model.encode([prompt])
-    bpm = int(reg.predict(prompt_enc)[0]) + 20
-    print(bpm, "bpm", "notes are", note_text)
-    audio = text_to_audio(note_text, bpm)
+    # prompt_enc = model.encode([prompt])
+    # bpm = int(reg.predict(prompt_enc)[0]) + 20
+    audio = text_to_audio(note_text, tempo)
     audio = np.array(audio.get_array_of_samples(), dtype=np.float32)
     return (96000, audio)
 
 iface = gr.Interface(
     fn=get_drummer_output,
-    inputs="text",
+    inputs=[
+        "text",
+        gr.Radio(["fast", "slow"], label="Tempo", default="fast"),
+    ]
     examples=[
         "hiphop groove 808",
         "rock metal",