Update app.py

app.py

@@ -29,12 +29,21 @@ def generate(prompts):
   images = generator(list(prompts)).images
   return [images]
 
-demo = gr.Interface(generate, 
-             "textbox", 
-             "image", 
-             batch=True, 
-             max_batch_size=4  # Set the batch size based on your CPU/GPU memory
-).queue()
-
+with gr.Blocks() as demo:
+    gr.HTML(
+
+    )
+
+    with gr.Row():
+        with gr.Row():
+            input_text = gr.Textbox(label="Input Prompt",  placeholder="", lines=1)
+            # Model selection dropdown
+            model_name1 = gr.Dropdown(
+                label="Choose Model",
+                choices=[m["name"] for m in models],
+                type="index",
+                value=current_model["name"],
+                interactive=True,
+                
 if __name__ == "__main__":
     demo.launch()