Update app.py

app.py

@@ -14,11 +14,11 @@ print(device)
 pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=auth_token, revision="fp16", torch_dtype=torch.float32)
 pipe = pipe.to(device)
 
-#stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
+stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
 
-#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 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 infer(prompt, samples, steps, scale, seed):        
@@ -39,7 +39,6 @@ def infer(prompt, samples, steps, scale, seed):
     #         images.append(image)
     return images
 
-
 block = gr.Blocks()
 
 with block:
@@ -87,4 +86,12 @@ with block:
             text,
         )
         
+        
+    output = gr.Textbox(label="Output Box")
+    sd_output = gr.Gallery().style(grid=2, height="auto")
+
+    submit_audio_button.click(fn=transcribe, inputs=audio_input, outputs=text_output)
+    text_output.change(fn=get_images, inputs=text_output, outputs=sd_output)
+    submit_button_text.click(fn=get_images, inputs=text_input, outputs=sd_output)
+        
 block.launch()