app.py

import gradio as gr
import os

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"
]

model_name = os.getenv("HUGGINGFACE_API_KEY")

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)