Update app.py

app.py

@@ -28,7 +28,16 @@ TORCH_DTYPE = os.getenv('TORCH_DTYPE', 'float16')  # Standard-Präzision
 
 # Modell laden
 device = "cuda" if torch.cuda.is_available() else "cpu"
-torch_dtype = torch.float16 if torch.cuda.is_available() and TORCH_DTYPE == 'float16' else torch.float32
+torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
+
+try:
+    pipe = StableDiffusionPipeline.from_pretrained(
+        MODEL_REPO, 
+        torch_dtype=torch_dtype
+    ).to(device)
+except Exception as e:
+    raise RuntimeError(f"Failed to load the model. Ensure the token has access to the repo. Error: {e}")
+
 
 
 try: