Update app.py

app.py

@@ -9,6 +9,7 @@ import base64
 import json
 import time
 
+
 class Prodia:
     def __init__(self, api_key=os.getenv("PRODIA_API_KEY"), base=None):
         self.base = base or "https://api.prodia.com/v1"
@@ -53,8 +54,10 @@ class Prodia:
 
         return response
 
+
 client = Prodia()
 
+
 def infer(source, target):
     if source_image is None or target_image is None:
         return
@@ -73,6 +76,7 @@ def infer(source, target):
 
     return res['imageUrl']
 
+
 def upload_image(file):
     files = {'file': open(file, 'rb')}
     img_id = requests.post(os.getenv("IMAGE_API_1"), files=files).json()['id']
@@ -87,15 +91,15 @@ def upload_image(file):
 
     return f"{os.getenv('IMAGE_API_1')}/{img_id}/{res.json()['attachments'][0]['filename']}"
 
+
 def image_to_base64(image: Image):
     buffered = BytesIO()
     image.save(buffered, format="PNG")
-
     img_str = base64.b64encode(buffered.getvalue())
-
     return img_str.decode('utf-8')
 
-with gr.Blocks() as demo:
+
+with gr.Blocks(description="Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.", theme="NoCrypt/miku") as demo:
     with gr.Column():
         gr.HTML("<h1><center>Face Swap</center></h1>")
 
@@ -107,12 +111,6 @@ with gr.Blocks() as demo:
             result = gr.Image()
             run_button = gr.Button("Swap Face", variant="primary")
 
-    gr.Examples(
-        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)