Update app.py

app.py

@@ -5,6 +5,8 @@ os.system("pip install -U huggingface_hub")
 os.system("pip install -U diffusers")
 '''
 
+os.system("git clone https://huggingface.co/warp-ai/wuerstchen")
+
 import sys
 
 import gradio as gr
@@ -16,8 +18,13 @@ from diffusers import AutoPipelineForText2Image
 from diffusers.pipelines.wuerstchen.pipeline_wuerstchen_prior import DEFAULT_STAGE_C_TIMESTEPS
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
+
+pipe = AutoPipelineForText2Image.from_pretrained("wuerstchen",
+    torch_dtype=torch.float32)
+'''
 pipe = AutoPipelineForText2Image.from_pretrained("warp-ai/wuerstchen",
     torch_dtype=torch.float32)
+'''
 pipe.to(device)
 
 pipe.safety_checker = None