Fix: Correct EndpointHandler class name and add robust error handling for model loading

handler.py

@@ -15,27 +15,33 @@ import random
 import math
 import re
 
-class DiffSketchEditHandler:
-    def __init__(self):
+class EndpointHandler:
+    def __init__(self, path=""):
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         self.model_id = "runwayml/stable-diffusion-v1-5"
         
-        # Initialize the diffusion pipeline
-        self.pipe = StableDiffusionPipeline.from_pretrained(
-            self.model_id,
-            torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-            safety_checker=None,
-            requires_safety_checker=False
-        ).to(self.device)
-        
-        # Use DDIM scheduler for better control
-        self.pipe.scheduler = DDIMScheduler.from_config(self.pipe.scheduler.config)
-        
-        # CLIP model for guidance
-        self.clip_model = self.pipe.text_encoder
-        self.clip_tokenizer = self.pipe.tokenizer
-        
-        print("DiffSketchEdit handler initialized successfully!")
+        try:
+            # Initialize the diffusion pipeline
+            self.pipe = StableDiffusionPipeline.from_pretrained(
+                self.model_id,
+                torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+                safety_checker=None,
+                requires_safety_checker=False
+            ).to(self.device)
+            
+            # Use DDIM scheduler for better control
+            self.pipe.scheduler = DDIMScheduler.from_config(self.pipe.scheduler.config)
+            
+            # CLIP model for guidance
+            self.clip_model = self.pipe.text_encoder
+            self.clip_tokenizer = self.pipe.tokenizer
+            
+            print("DiffSketchEdit handler initialized successfully!")
+        except Exception as e:
+            print(f"Warning: Could not load diffusion model: {e}")
+            self.pipe = None
+            self.clip_model = None
+            self.clip_tokenizer = None
 
     def __call__(self, inputs: Union[str, Dict[str, Any]]) -> Image.Image:
         """