Update app.py

app.py

@@ -7,6 +7,9 @@ from pulid import attention_processor as attention
 from pulid.pipeline import PuLIDPipeline
 from pulid.utils import resize_numpy_image_long, seed_everything
 
+import boto3
+from botocore.exceptions import ClientError
+
 torch.set_grad_enabled(False)
 
 pipeline = PuLIDPipeline()
@@ -182,7 +185,14 @@ with gr.Blocks(title="AI headshot Generator", css=".gr-box {border-color: #8136e
 
     output = gr.Gallery(label='Output', elem_id="gallery")
     intermediate_output = gr.Gallery(label='DebugImage', elem_id="gallery", visible=False)
-    submit = gr.Button("Generate")
-    submit.click(fn=run, inputs=['textbox', 'textbox', 'textbox'], outputs=[output, intermediate_output])
+    # submit = gr.Button("Generate")
+    # submit.click(fn=run, inputs=['textbox', 'textbox', 'textbox'], outputs=[output, intermediate_output])
+
+demo = gr.Interface(
+    run,
+    inputs=['textbox', 'textbox', 'textbox'], 
+    outputs=[output, intermediate_output],
+    title="AI Headshot Diffusor"
+)
 
 demo.launch()