Update app.py

app.py

@@ -6,7 +6,7 @@ import gfpgan
 import tempfile
 import time
 import gradio as gr
-
+from gradio.themes.utils.theme_dropdown import create_theme_dropdown # noqa: F401
 
 class Predictor:
     def __init__(self):
@@ -69,17 +69,24 @@ class Predictor:
 predictor = Predictor()
 title = "🧸 Auto adjust Models 🧸"
 
-# Create Gradio Interface
-iface = gr.Interface(
-    fn=predictor.predict,
-    inputs=[
-        gr.Image(type="filepath", label="Target Image"),
-        gr.Image(type="filepath", label="Swap Image")
-    ],
-    outputs=gr.Image(type="filepath", label="Result"),
-    title=title,
-    examples=[["input.jpg", "swap img.jpg"]]
-)
+# Create theme dropdown
+dropdown, js = create_theme_dropdown()
+
+# Create Gradio Interface with the specified theme
+with gr.Blocks(theme='HaleyCH/HaleyCH_Theme') as demo:
+    with gr.Row(equal_height=True):
+        with gr.Column(scale=10):
+            iface = gr.Interface(
+                fn=predictor.predict,
+                inputs=[
+                    gr.Image(type="filepath", label="Target Image"),
+                    gr.Image(type="filepath", label="Swap Image")
+                ],
+                outputs=gr.Image(type="filepath", label="Result"),
+                title=title,
+                examples=[["input.jpg", "swap img.jpg"]]
+            )
+    demo.append(dropdown)
 
 # Launch the Gradio Interface
-iface.launch()
+demo.launch()