input chn

check_app.py

@@ -89,13 +89,13 @@ def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, s
     return image
 
 @spaces.GPU(duration=170)
-def create_pipeline_logic(prompt_text, model_name, config):
+def create_pipeline_logic(prompt_text, model_name):
     print(f"starting {model_name}")
     progress = gr.Progress()
     num_steps = 30
     guidance_scale = 7.5  # Example guidance scale, can be adjusted per model
     seed = 42
-    
+    config = MODEL_CONFIGS[model_name]
     pipe_class = config["pipeline_class"]
     pipe = pipe_class.from_pretrained(
         config["repo_id"],
@@ -119,7 +119,7 @@ def main():
                 output = gr.Textbox(label="Status")
                 img = gr.Image(label=model_name, height=300)
 
-                button.click(fn=create_pipeline_logic, inputs=[prompt_text, model_name, config], outputs=[output, img])
+                button.click(fn=create_pipeline_logic, inputs=[prompt_text, gr.Text(value= model_name,visible=False)], outputs=[output, img])
 
     app.launch()