Rename app.py to _app.py

app.py → _app.py

@@ -5,6 +5,16 @@ import torch
 import gradio as gr
 from diffusers import StableDiffusionPipeline
 import paramiko
+from huggingface_hub import login
+
+# Hugging Face Token
+HF_TOKEN = os.getenv('HUGGING_FACE_HUB_TOKEN', '').strip()
+
+if not HF_TOKEN:
+    raise ValueError("HUGGING_FACE_HUB_TOKEN is not set. Please set the token as an environment variable.")
+
+# Hugging Face Login
+login(token=HF_TOKEN)
 
 # Konfiguration
 STORAGE_DOMAIN = os.getenv('STORAGE_DOMAIN', '').strip()  # SFTP Server Domain
@@ -16,7 +26,11 @@ STORAGE_SECRET = os.getenv('STORAGE_SECRET', '').strip()  # Secret Token
 # Modell laden
 device = "cuda" if torch.cuda.is_available() else "cpu"
 repo = "stabilityai/stable-diffusion-3-medium-diffusers"
-pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=torch.float16).to(device)
+
+try:
+    pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=torch.float16).to(device)
+except Exception as e:
+    raise RuntimeError(f"Failed to load the model. Ensure the token has access to the repo. Error: {e}")
 
 # Maximalwerte
 MAX_SEED = np.iinfo(np.int32).max