Update app.py

app.py

@@ -26,8 +26,8 @@ for model_path in models:
 
 def send_it_idx(idx):
     def send_it_fn(prompt):
-        output = (model_functions.get(idx) or model_functions.get(1)).predict(prompt)
-        return output
+        model = model_functions.get(idx, model_functions.get(1))
+        return model.predict(prompt)
     return send_it_fn
 
 def get_prompts(prompt_text):
@@ -71,7 +71,6 @@ with gr.Blocks(title="SD Models") as my_interface:
     end_box = gr.Number(visible=False)
     tog_box = gr.Textbox(value=0, visible=False)
 
-    # Define event handlers within the Blocks context
     def main_function(prompt):
         return {idx: run_model(prompt, idx) for idx in range(1, len(models) + 1)}
 
@@ -104,5 +103,4 @@ with gr.Blocks(title="SD Models") as my_interface:
         outputs=tog_box
     )
 
-    # Launch the interface
     my_interface.launch(max_threads=10)