Create app.py

app.py

@@ -0,0 +1,37 @@
+import gradio as gr
+from huggingface_hub import InferenceClient
+
+# List of models and their corresponding names
+models = [
+    "CompVis/stable-diffusion-v1-4",
+    "runwayml/stable-diffusion-v1-5",
+    "hakurei/waifu-diffusion",
+    "prompthero/openjourney",
+    "jacek-huszar/$(IMG2TEXT_MODEL_NAME)",
+    "cjwbw/canvae-text-to-image",
+    "tdbooth/hollow-kingdom-ddim",
+    "kuprel/min-dalle",
+    "osanseviero/stable-diffusion-webui",
+    "lambdal/text-to-image"
+]
+
+def generate_image(prompt, model_name):
+    client = InferenceClient(model_name)
+    response = client.text_to_image(prompt)
+    return response.image
+
+# Create Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("## Text-to-Image Generation with Hugging Face Models")
+    with gr.Row():
+        with gr.Column():
+            model_dropdown = gr.Dropdown(models, label="Select Model")
+            prompt_input = gr.Textbox(label="Enter Text Prompt")
+            generate_button = gr.Button("Generate Image")
+        with gr.Column():
+            output_image = gr.Image(label="Generated Image")
+    
+    generate_button.click(generate_image, inputs=[prompt_input, model_dropdown], outputs=output_image)
+
+# Launch the interface
+demo.launch()