Update app.py

app.py

@@ -339,22 +339,27 @@ td {
 <center><img src='https://api.infinitescript.com/badgen/count?name=sczhou/CodeFormer&ltext=Visitors&color=6dc9aa' alt='visitors'></center>
 """
 
-demo = gr.Interface(
-    inference, [
-        gr.Image(type="filepath", label="Input"),
-        gr.Checkbox(value=True, label="Pre_Face_Align"),
-        gr.Checkbox(value=True, label="Background_Enhance"),
-        gr.Checkbox(value=True, label="Face_Upsample"),
-        gr.Number(value=2, label="Rescaling_Factor (up to 4)"),
-        gr.Slider(0, 1, value=0.5, step=0.01, label='Codeformer_Fidelity (0 for better quality, 1 for better identity)')
-    ], [
-        gr.Image(type="filepath", label="Output"),
-        gr.Markdown("**Please download the output within 30 seconds.**")
-    ],
-    title=title,
-    description=description,
-    article=article,       
-    examples=[
+with gr.Blocks() as demo:
+    gr.Markdown(title)
+    gr.Markdown(description)
+    with gr.Box():
+        with gr.Column():
+            input_img = gr.Image(type="filepath", label="Input")
+            face_align = gr.Checkbox(value=True, label="Pre_Face_Align")
+            background_enhance = gr.Checkbox(value=True, label="Background_Enhance")
+            face_enhance = gr.Checkbox(value=True, label="Face_Upsample")
+            upscale_factor = gr.Number(value=2, label="Rescaling_Factor (up to 4)")
+            codeformer_fidelity = gr.Slider(0, 1, value=0.5, step=0.01, label='Codeformer_Fidelity (0 for better quality, 1 for better identity)')
+            submit = gr.Button('Enhance Image')
+        with gr.Column():
+            output_img = gr.Image(type="filepath", label="Output").style(height='auto')
+            note = gr.Markdown("**Please download the output within 30 seconds.**")
+            
+    inps = [input_img, face_align, background_enhance, face_enhance, upscale_factor, codeformer_fidelity]
+    outs = [output_img, note]
+    submit.click(fn=inference, inputs=inps, outputs=outs)
+            
+    ex = gr.Examples([
         ['01.png', True, True, True, 2, 0.7],
         ['02.jpg', True, True, True, 2, 0.7],
         ['03.jpg', True, True, True, 2, 0.7],
@@ -362,9 +367,11 @@ demo = gr.Interface(
         ['05.jpg', True, True, True, 2, 0.1],
         ['06.png', False, True, True, 1, 0.5]
       ],
-    concurrency_limit=2
-    )
-
+        inputs=inps,
+        cache_examples=False)
+    
+    gr.Markdown(article)
+    
 DEBUG = os.getenv('DEBUG') == '1'
-# demo.launch(debug=DEBUG)
-demo.launch(debug=DEBUG, share=True, cache_examples=False)
+demo.queue(api_open=False, concurrency_count=2, max_size=10)
+demo.launch(debug=DEBUG)