Update app.py

app.py

@@ -21,7 +21,7 @@ for model_path in models:
     except Exception as error:
         def the_fn(txt):
             return None
-        model_functions[model_idx] = gr.Interface(fn=the_fn, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Image())
+        model_functions[model_idx] = gr.Interface(fn=the_fn, inputs=gr.Textbox(), outputs=gr.Image())
     model_idx += 1
 
 def send_it_idx(idx):
@@ -58,19 +58,18 @@ def clear_fn(*args):
 def run_model(prompt, model_idx):
     return send_it_idx(model_idx)(prompt)
 
-# Create UI components
 def build_ui():
-    primary_prompt = gr.inputs.Textbox(label="Prompt", default="")
+    primary_prompt = gr.Textbox(label="Prompt", default="")
     run = gr.Button("Run")
     clear_btn = gr.Button("Clear")
 
     sd_outputs = {}
     for idx, model_path in enumerate(models, start=1):
-        sd_outputs[idx] = gr.outputs.Image(label=model_path)
+        sd_outputs[idx] = gr.Image(label=model_path)
 
-    start_box = gr.inputs.Number(visible=False)
-    end_box = gr.inputs.Number(visible=False)
-    tog_box = gr.inputs.Textbox(default=0, visible=False)
+    start_box = gr.Number(visible=False)
+    end_box = gr.Number(visible=False)
+    tog_box = gr.Textbox(default=0, visible=False)
 
     return primary_prompt, run, clear_btn, sd_outputs, start_box, end_box, tog_box