Update app.py

app.py

@@ -4,6 +4,7 @@ import json
 import PIL.Image
 from io import BytesIO
 import os
+
 def generate_image(prompt, negative_prompt, scheduler, steps, width, height, restore_faces):
     # Define the API endpoint
     apiUrl = os.getenv("API_URL")
@@ -74,16 +75,16 @@ def generate_image(prompt, negative_prompt, scheduler, steps, width, height, res
         image = PIL.Image.open(BytesIO(image_response.content))
 
         return image
-    else:
+   else:
         raise Exception("API request failed with status code " + str(response.status_code))
 
 # Define the Gradio interface
 iface = gr.Interface(
     fn=generate_image, 
     inputs=[
-        gr.inputs.Textbox(label="Prompt"),
-        gr.inputs.Textbox(label="Negative Prompt"),
-        gr.inputs.Dropdown(choices=[
+        gr.components.Textbox(label="Prompt"),
+        gr.components.Textbox(label="Negative Prompt"),
+        gr.components.Dropdown(choices=[
             "Euler a",
             "Euler",
             "LMS",
@@ -104,12 +105,12 @@ iface = gr.Interface(
             "DDIM",
             "PLMS"
         ], label="Scheduler", default="Euler a"),
-        gr.inputs.Slider(minimum=10, maximum=60, default=30, label="Steps"),
-        gr.inputs.Slider(minimum=512, maximum=1500, default=768, label="Width"),
-        gr.inputs.Slider(minimum=512, maximum=1500, default=768, label="Height"),
-        gr.inputs.Checkbox(label="Restore Faces")
+        gr.components.Slider(minimum=10, maximum=60, default=30, label="Steps"),
+        gr.components.Slider(minimum=512, maximum=1500, default=768, label="Width"),
+        gr.components.Slider(minimum=512, maximum=1500, default=768, label="Height"),
+        gr.components.Checkbox(label="Restore Faces")
     ], 
-    outputs="image",
+    outputs=gr.components.Image(),
     title="Freedom Demonstration",
     description = """
 Testing environment for the Freedom Model. Finetuned model of SD 2.1 768X produced by <a href='https://twitter.com/artificialguybr' target='_blank'>@artificialguybr</a>.<br>
@@ -118,8 +119,8 @@ You can find example prompts <a href='LINK_TO_EXAMPLE_PROMPTS' target='_blank'>h
 Demonstration running on the <a href='LINK_TO_MAKEAI.RUN_API' target='_blank'>makeai.run API</a>.<br>
 Thanks to <a href='LINK_TO_REDMOND.AI' target='_blank'>Redmond.ai</a> for providing GPU Time and sponsoring this model.
 """,
-    allow_flagging=False
+    allow_flagging='never'
 )
 
 # Launch the app
-iface.launch()
+iface.launch()