Update app.py

app.py

@@ -3,7 +3,8 @@ import gradio as gr
 proc1 = gr.Interface.load("models/prompthero/funko-diffusion",live=True,preprocess=True, postprocess=False)
 
 def send_it1(inputs,proc1=proc1):
-    output1=proc1(inputs)
+    output1= await proc1(inputs)
+    print(output1)
     return output1
 
 with gr.Blocks() as myface:
@@ -29,11 +30,9 @@ with gr.Blocks() as myface:
                     full_width=False,
                 )
 
-        gallery = gr.Gallery(
-            label="Generated images", show_label=False, elem_id="gallery"
-        ).style(grid=[2], height="auto")
+    
 
-        btn.click(send_it1, inputs=[text], outputs=[gallery], postprocess=False, api_name="text")
+        btn.click(send_it1, inputs=[text], outputs=gr.outputs.Image(type="auto", label="Generated Image"), postprocess=False, api_name="text")
 
 if __name__ == "__main__":
      myface.launch()