Update app.py

app.py

@@ -9,8 +9,12 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 num_images_per_prompt = 1
 print(f"Running on: {device}")
 
-# Assuming you've corrected the model paths as per previous discussions
-prior = StableCascadePriorPipeline.from_pretrained("./SomniumSC-v1.1", torch_dtype=torch.bfloat16).to(device)
+try:
+    prior = StableCascadePriorPipeline.from_pretrained("Ketengan-Diffusion/SomniumSC-v1.1", torch_dtype=torch.bfloat16).to(device)
+except OSError:
+    print("Failed to load model from Hugging Face Model Hub. Loading from local path instead.")
+    prior = StableCascadePriorPipeline.from_pretrained("./SomniumSC-v1.1", torch_dtype=torch.bfloat16).to(device)
+    
 decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.float16).to(device)
 
 deafult_negative = "lowres"