Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import os
 
+api_key = os.environ.get("HUGGINGFACE_API_KEY")
+
 model_list = [
     "stabilityai/stable-diffusion-xl-base-0.9",
     "stabilityai/stable-diffusion-2-1",
@@ -17,14 +19,26 @@ model_list = [
     "stabilityai/stable-diffusion-2-1-unclip-small"
 ]
 
-model_name = os.getenv("HUGGINGFACE_API_KEY")
+default_model = "stabilityai/stable-diffusion-2"
+model_name = gr.inputs.Dropdown(model_list, label="Select Model", default=default_model)
+model = None
+
+def load_model(model_name):
+    global model
+    model = gr.Interface.load(model_name, api_key=api_key)
 
-if model_name is None or model_name not in model_list:
-    print("Invalid Hugging Face model name provided.")
-    exit()
+def predict(inputs):
+    return model.predict(inputs)
 
-model_path = f"models/{model_name}"
+iface = gr.Interface(
+    fn=predict,
+    inputs=model,
+    outputs="text",
+    capture_session=True,
+    title="Model Selection App",
+    description="Choose a model and input data to make predictions."
+)
 
-title = "Mujo kuje AI po mjesecu"
+load_model(default_model)  # Load the default model initially
 
-gr.Interface.load(model_path).launch(title=title)
+iface.launch()