Update app.py

app.py

@@ -15,34 +15,38 @@ models = [
     {"name": "Anything-v4.5-pruned", "url": "anything-v4.5-pruned.ckpt [65745d25]"},
 ]
 
- current_model = models[0]
+current_model = models[0]
 
-response = requests.post(url, json=payload, headers=headers)
-generator = "timeless-1.0.ckpt [7c4971d4]"
+models2 = []
+for model in models:
+    model_url = f"models/{model['url']}"
+    loaded_model = gr.Interface.load(model_url, live=True, preprocess=True)
+    models2.append(loaded_model)
 
-print(response.text)
 
-response = requests.post(url, headers=headers)
+def text_it(inputs, text_gen=text_gen):
+    return text_gen(inputs)
 
-#generator = DiffusionPipeline.from_pretrained("CompVis/ldm-text2im-large-256")
-
-def generate(prompts):
-  images = generator(list(prompts)).images
-  return [images]
 
 def set_model(current_model_index):
     global current_model
     current_model = models[current_model_index]
     return gr.update(value=f"{current_model['name']}")
 
-with gr.Blocks() as demo:
+
+def send_it(inputs, model_choice):
+    proc = models2[model_choice]
+    return proc(inputs)
+
+
+with gr.Blocks() as myface:
     gr.HTML(
 
     )
 
     with gr.Row():
         with gr.Row():
-            input_text = gr.Textbox(label="Input Prompt",  placeholder="", lines=1)
+            input_text = gr.Textbox(label="Iput Prompt",  placeholder="", lines=1)
             # Model selection dropdown
             model_name1 = gr.Dropdown(
                 label="Choose Model",
@@ -51,5 +55,9 @@ with gr.Blocks() as demo:
                 value=current_model["name"],
                 interactive=True,
             )
+        with gr.Row():
+            see_prompts = gr.Button("Generate Prompts")
+            run = gr.Button("Generate Images", variant="primary")
+
 if __name__ == "__main__":
     demo.launch()