Update app.py

app.py

@@ -25,7 +25,7 @@ from diffusers.pipelines.wuerstchen.pipeline_wuerstchen_prior import DEFAULT_STA
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
-
+assert os.path.exists("wuerstchen")
 pipe = AutoPipelineForText2Image.from_pretrained(Path("wuerstchen"), local_files_only = True,
     torch_dtype=torch.float32)
 '''