Update app.py

app.py

@@ -1,25 +1,30 @@
 import gradio as gr
-from gradio import components
+import os
 
-models = [
-    "models/stabilityai/stable-diffusion-xl-base-0.9",
-    "models/stabilityai/stable-diffusion-2-1",
-    "models/stabilityai/stable-diffusion-xl-refiner-0.9",
-    "models/stabilityai/stable-diffusion-2-1-base",
-    "models/stabilityai/stable-diffusion-2",
-    "models/stabilityai/stable-diffusion-2-inpainting",
-    "models/stabilityai/stable-diffusion-x4-upscaler",
-    "models/stabilityai/stable-diffusion-2-depth",
-    "models/stabilityai/stable-diffusion-2-base",
-    "models/stabilityai/stable-diffusion-2-1-unclip",
-    "models/helenai/stabilityai-stable-diffusion-2-1-base-ov",
-    "models/helenai/stabilityai-stable-diffusion-2-1-ov",
-    "models/stabilityai/stable-diffusion-2-1-unclip-small"
+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"
 ]
 
-def choose_model(model_name):
-    gr.Interface.load(model_name).launch()
+model_name = os.getenv("HUGGINGFACE_API_KEY")
 
-model_dropdown = components.Dropdown(models)
-interface = gr.Interface(fn=choose_model, inputs=model_dropdown, outputs=None)
-interface.launch()
+if model_name is None or model_name not in model_list:
+    print("Invalid Hugging Face model name provided.")
+    exit()
+
+model_path = f"models/{model_name}"
+
+title = "Mujo kuje AI po mjesecu"
+
+gr.Interface.load(model_path).launch(title=title)