Update app.py

app.py

@@ -10,15 +10,17 @@ import base64
 client = Prodia()
 
 
-def run_face_swap(source, target):
+def infer(source, target):
     if source_image is None or target_image is None:
         return
 
     source_url = upload_image(source)
     target_url = upload_image(target)
-    
+
     job = client.faceswap(source_url=source_url, target_url=target_url)
-    res = client.wait(job)
+    res = client.wait(job, raise_on_fail=False)
+    if res.failed:
+        return
 
     return res.image_url
 
@@ -61,6 +63,18 @@ with gr.Blocks() as demo:
             result = gr.Image()
             run_button = gr.Button("Swap Faces", variant="primary")
 
-    run_button.click(fn=run_face_swap, inputs=[source_image, target_image], outputs=result)
+    gr.Examples(
+        examples=[
+            [
+                "/examples/example1.jpg",
+                "/examples/example2.jpg"
+            ]
+        ],
+        fn=infer,
+        inputs=[source_image, target_image],
+        outputs=[result]
+    )
+
+    run_button.click(fn=infer, inputs=[source_image, target_image], outputs=[result])
 
 demo.queue(max_size=20, api_open=False).launch(show_api=False, max_threads=400)