Update app.py

app.py

@@ -4,38 +4,40 @@ import os
 
 api_key = os.environ.get("HUGGINGFACE_API_KEY")
 
-model_list = [
-    "stabilityai/stable-diffusion-xl-base-0.9",
-    "stabilityai/stable-diffusion-2-1",
-    "stabilityai/stable-diffusion-xl-refiner-0.9",
-    "stabilityai/stable-diffusion-2-1-base",
-    "stabilityai/stable-diffusion-2",
-    "stabilityai/stable-diffusion-2-inpainting",
-    "stabilityai/stable-diffusion-x4-upscaler",
-    "stabilityai/stable-diffusion-2-depth",
-    "stabilityai/stable-diffusion-2-base",
-    "stabilityai/stable-diffusion-2-1-unclip",
-    "helenai/stabilityai-stable-diffusion-2-1-base-ov",
-    "helenai/stabilityai-stable-diffusion-2-1-ov",
-    "stabilityai/stable-diffusion-2-1-unclip-small"
-]
-
-default_model = "stabilityai/stable-diffusion-2"
-model_name = gr.inputs.Dropdown(choices=model_list, label="Select Model", default=default_model)
-
-def generate_image(text, default_model):
-    model = gr.load(default_model, source="huggingface", api_key=api_key)
-    return model.predict(text)
-
-input_text = gr.inputs.Textbox(label="Input Text")
-output_image = comp.Image(label="Generated Image")
-
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=[input_text, default_model],
-    outputs=output_image,
-    title="Text to Image Generation",
-    description="Generate an image from input text using a Hugging Face model."
-)
-
-iface.launch()
+#model_list = [
+#    "stabilityai/stable-diffusion-xl-base-0.9",
+#    "stabilityai/stable-diffusion-2-1",
+#    "stabilityai/stable-diffusion-xl-refiner-0.9",
+#    "stabilityai/stable-diffusion-2-1-base",
+#    "stabilityai/stable-diffusion-2",
+#    "stabilityai/stable-diffusion-2-inpainting",
+#    "stabilityai/stable-diffusion-x4-upscaler",
+#    "stabilityai/stable-diffusion-2-depth",
+#    "stabilityai/stable-diffusion-2-base",
+#    "stabilityai/stable-diffusion-2-1-unclip",
+#    "helenai/stabilityai-stable-diffusion-2-1-base-ov",
+#    "helenai/stabilityai-stable-diffusion-2-1-ov",
+#    "stabilityai/stable-diffusion-2-1-unclip-small"
+#]
+
+#default_model = "stabilityai/stable-diffusion-2"
+#model_name = gr.inputs.Dropdown(choices=model_list, label="Select Model", default=default_model)
+
+#def generate_image(text, default_model):
+#    model = gr.load(default_model, source="huggingface", api_key=api_key)
+#    return model.predict(text)
+
+#input_text = gr.inputs.Textbox(label="Input Text")
+#output_image = comp.Image(label="Generated Image")
+
+#iface = gr.Interface(
+#    fn=generate_image,
+#    inputs=[input_text, default_model],
+#    outputs=output_image,
+#    title="Text to Image Generation",
+#    description="Generate an image from input text using a Hugging Face model."
+#)
+
+#iface.launch()
+
+gr.Interface.load("models/stabilityai/stable-diffusion-2-1").launch()