Update app.py (#5)

- Update app.py (83a45d3c63be98c9a3088f2921b1e63194f1934f)


Co-authored-by: Aditya Deshmukh <[email protected]>

app.py

@@ -261,31 +261,32 @@ def custom_infer(input_image_path,
         a, b = predict(input_dict, prompt, fitting_degree, ddim_steps, scale, seed, negative_prompt, task, left_expansion_ratio, right_expansion_ratio, top_expansion_ratio, bottom_expansion_ratio)
 
     return a, b
+import gradio as gr
 
-# Create the Gradio interface
+# Define the Gradio interface using the new version
 inputs = [
-    gr.inputs.Image(label="Input Image", type="filepath"),
-    gr.inputs.Image(label="Input Mask (optional)", type="filepath"),
-    gr.inputs.Textbox(label="Prompt", default="A beautiful landscape"),
-    gr.inputs.Slider(label="Fitting Degree", minimum=1, maximum=20, value=7, step=1),
-    gr.inputs.Slider(label="DDIM Steps", minimum=10, maximum=100, value=20, step=1),
-    gr.inputs.Slider(label="Scale", minimum=1, maximum=20, value=7.5, step=0.1),
-    gr.inputs.Slider(label="Use Seed", minimum=0, maximum=1300000,value=143,step=1),
-    gr.inputs.Textbox(label="Negative Prompt", default="blur, low quality"),
-    gr.inputs.Radio(label="Task", choices=["text-guided", "image-outpainting"], value="image-outpainting"),
-    gr.inputs.Slider(label="Left Expansion Ratio", minimum=0, maximum=2, value=0.2, step=0.01),
-    gr.inputs.Slider(label="Right Expansion Ratio", minimum=0, maximum=2, value=0.2, step=0.01),
-    gr.inputs.Slider(label="Top Expansion Ratio", minimum=0, maximum=2, value=0.2, step=0.01),
-    gr.inputs.Slider(label="Bottom Expansion Ratio", minimum=0, maximum=2, value=0.2, step=0.01)
+    gr.Image(label="Input Image", type="filepath"),
+    gr.Image(label="Input Mask (optional)", type="filepath"),
+    gr.Textbox(label="Prompt", value="A beautiful landscape"),
+    gr.Slider(label="Fitting Degree", minimum=1, maximum=20, value=7, step=1),
+    gr.Slider(label="DDIM Steps", minimum=10, maximum=100, value=20, step=1),
+    gr.Slider(label="Scale", minimum=1, maximum=20, value=7.5, step=0.1),
+    gr.Slider(label="Use Seed", minimum=0, maximum=1300000, value=143, step=1),
+    gr.Textbox(label="Negative Prompt", value="blur, low quality"),
+    gr.Radio(label="Task", choices=["text-guided", "image-outpainting"], value="image-outpainting"),
+    gr.Slider(label="Left Expansion Ratio", minimum=0, maximum=2, value=0.2, step=0.01),
+    gr.Slider(label="Right Expansion Ratio", minimum=0, maximum=2, value=0.2, step=0.01),
+    gr.Slider(label="Top Expansion Ratio", minimum=0, maximum=2, value=0.2, step=0.01),
+    gr.Slider(label="Bottom Expansion Ratio", minimum=0, maximum=2, value=0.2, step=0.01)
 ]
 
 outputs = [
-    gr.outputs.Image(label="Output Image"),
-    gr.outputs.Textbox(label="Output Text")
+    gr.Image(label="Output Image"),
+    gr.Textbox(label="Output Text")
 ]
 
 # Create the Gradio interface
-demo = gr.Interface(fn=custom_infer, inputs=inputs, outputs=outputs, title="Custom Infer")
+demo = gr.Interface(fn=custom_infer, inputs=inputs, outputs=outputs, title="Inference")
 
 demo.queue(concurrency_count=1, max_size=1, api_open=True)
 demo.launch(show_api=True)