Update app.py

app.py

@@ -10,13 +10,10 @@ from diffusers import StableDiffusionImg2ImgPipeline
 
 device = "cpu"
 MODEL_ID = "Hyeon2/riffusion-musiccaps"
-pipe = StableDiffusionPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.float16)
+pipe = StableDiffusionPipeline.from_pretrained(MODEL_ID)
 pipe = pipe.to(device)
 
 def predict(prompt, duration):
-    return classic(prompt, duration)
-
-def classic(prompt, duration):
     if duration == 5:
         width_duration=512
     else:
@@ -26,7 +23,7 @@ def classic(prompt, duration):
     wav = wav_bytes_from_spectrogram_image(spec)
     with open("output.wav", "wb") as f:
         f.write(wav[0].getbuffer())
-    return spec, 'output.wav'
+    return spec, "output.wav"
 
 title = """
     <div style="text-align: center; max-width: 500px; margin: 0 auto;">
@@ -114,9 +111,8 @@ with gr.Blocks(css="style.css") as demo:
         
         gr.HTML(title)
         
-        prompt_input = gr.Textbox(placeholder="A LoFi beat", label="Musical prompt", elem_id="prompt-in")
-        with gr.Row():
-            negative_prompt = gr.Textbox(label="Negative prompt")
+        with gr.column():
+            prompt_input = gr.Textbox(placeholder="A LoFi beat", label="Musical prompt", elem_id="prompt-in")
             duration_input = gr.Slider(label="Duration in seconds", minimum=5, maximum=10, step=1, value=8, elem_id="duration-slider")
             
         send_btn = gr.Button(value="Get a new spectrogram!", elem_id="submit-btn")