Update app.py

app.py

@@ -8,14 +8,25 @@ hf_token = os.environ.get('HF_TOKEN')
 sdxl_client = Client("https://fffiloni-sdxl-dpo.hf.space/")
 faceswap_client = Client("https://fffiloni-deepfakeai.hf.space/", hf_token=hf_token)
 
-def infer(portrait_in, prompt_in):
-    # Generate Image from SDXL
-    gr.Info("Generating SDXL image first ...")
+def get_sdxl(prompt_in):
     sdxl_result = sdxl_client.predict(
         prompt_in,
         api_name="/infer"
     )
+    return sdxl_result
 
+def infer(portrait_in, prompt_in):
+    # Generate Image from SDXL
+    gr.Info("Generating SDXL image first ...")
+    # Keep trying the operation until it succeeds without raising an exception
+    while True:
+        try:
+            sdxl_result = get_sdxl(prompt_in)
+            break  # Exit the while loop if the operation succeeded
+        except Exception as e:
+            print(f"Operation failed with error: {e}")
+            time.sleep(3)  # Wait for 5 seconds before attempting again
+    
     unique_id = str(uuid.uuid4())
     
     # Face Swap
@@ -45,10 +56,11 @@ with gr.Blocks(css=css) as demo:
         <h2 style="text-align: center;">Portrait Maker</h2>
         """)
         with gr.Row():
-            with gr.Column():
-                portrait_in = gr.Image(label="Your face portrait", type="filepath")
-                prompt_in = gr.Textbox(label="Prompt to desired portrait using your own face")
-                submit_btn = gr.Button("Submit")
+            with gr.Group():
+                with gr.Column():
+                    portrait_in = gr.Image(label="Your face portrait", type="filepath")
+                    prompt_in = gr.Textbox(label="Prompt to desired portrait using your own face")
+                    submit_btn = gr.Button("Submit")
             with gr.Column():
                 result = gr.Image(label="Result")