Update app.py

app.py

@@ -14,12 +14,6 @@ device = "cuda"
 
 spectro_from_wav = gr.Interface.load("spaces/fffiloni/audio-to-spectrogram")
 
-def predict(prompt, negative_prompt, audio_input, duration):
-    if audio_input == None :
-        return classic(prompt, negative_prompt, duration)
-    else :
-        return style_transfer(prompt, negative_prompt, audio_input)
-
 def classic(model_input, prompt, negative_prompt, duration):
     pipe = StableDiffusionPipeline.from_pretrained(model_input, torch_dtype=torch.float16)
     pipe = pipe.to(device)
@@ -196,7 +190,7 @@ with gr.Blocks(css=css) as demo:
         
         gr.HTML(article)
     
-    send_btn.click(predict, inputs=[model_input, prompt_input, negative_prompt, audio_input, duration_input], outputs=[spectrogram_output, sound_output, share_button, community_icon, loading_icon])
+    send_btn.click(classic, inputs=[model_input, prompt_input, negative_prompt, audio_input, duration_input], outputs=[spectrogram_output, sound_output, share_button, community_icon, loading_icon])
     share_button.click(None, [], [], _js=share_js)
 
 demo.queue(max_size=250).launch(debug=True)