Update app.py

app.py

@@ -14,7 +14,12 @@ if torch.cuda.is_available():
 else:
     torch_dtype = torch.float32
 
-pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
+pipe = StableDiffusionXLPipeline.from_single_file(
+    ckpt_path,
+    use_safetensors=True,
+    torch_dtype=torch.float16,
+)
+
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 pipe.scheduler.register_to_config(
     prediction_type="v_prediction",