Update handler.py

handler.py

@@ -2,14 +2,19 @@ from diffusers import DiffusionPipeline
 import torch
 
 class EndpointHandler:
-    def __init__(self, path=""):
+    def __init__(self, path="Texttra/Cityscape_Studio"):
         print("Loading pipeline...")
-        self.pipe = DiffusionPipeline.from_pretrained(
-            path,
-            torch_dtype=torch.float16,
-            revision="fp16",
-            use_safetensors=True
-        ).to("cuda")
+        try:
+            self.pipe = DiffusionPipeline.from_pretrained(
+                path,
+                torch_dtype=torch.float16,
+                revision="fp16",
+                use_safetensors=True
+            ).to("cuda")
+            print("Pipeline loaded successfully.")
+        except Exception as e:
+            print(f"Error loading model: {e}")
+            raise
 
     def __call__(self, data):
         inputs = data.pop("inputs", data)