Update app.py

app.py

@@ -13,14 +13,7 @@ def get_images(prompt):
     gallery_dir = stable_diffusion(prompt, fn_index=2)
     return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
 
-def magic_whisper_to_sd(audio, guidance_scale, nb_iterations, seed):
-    
-    whisper_results = translate_better(audio)
-    prompt = whisper_results[1]
-    images = get_images(prompt)
-    
-    return whisper_results[0], whisper_results[1], images
-    
+
 def translate_better(audio):
     print("""
     —
@@ -88,17 +81,13 @@ with gr.Blocks() as demo:
             ) 
         
         sd_output = gr.Gallery().style(grid=2, height="auto")
-        audio_r_direct_sd.click(magic_whisper_to_sd, 
+        audio_r_translate.click(translate_better, 
                                 inputs = [
-                                    record_input, 
-                                    guidance_scale, 
-                                    nb_iterations, 
-                                    seed
+                                    record_input
                                 ], 
                                 outputs = [
                                     transcripted_output,
                                     translated_output,
-                                    sd_output
                                 ])
         audio_r_direct_sd.click(get_images, 
                               inputs = [