Update app.py (#6)

- Update app.py (8ca1d82e0adc7c0a556dd258cd292dfb48b6086a)


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

app.py

@@ -260,7 +260,7 @@ def custom_infer(input_image_path,
         input_dict = {"image": image}
         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
+    return a[0]
 import gradio as gr
 
 # Define the Gradio interface using the new version
@@ -281,8 +281,7 @@ inputs = [
 ]
 
 outputs = [
-    gr.Image(label="Output Image"),
-    gr.Textbox(label="Output Text")
+    gr.Image(label="Output Image")
 ]
 
 # Create the Gradio interface