Update controlnet_module.py

controlnet_module.py

@@ -35,6 +35,55 @@ class ControlNetProcessor:
         self.midas_transform = None
         self.sam_predictor = None
         self.sam_initialized = False
+
+
+    # In controlnet_module.py - Ersetze die _lazy_load_sam() Funktion
+from transformers import Sam2Model, Sam2Processor
+
+
+    # In controlnet_module.py - Ersetze die _lazy_load_sam() Funktion
+from transformers import Sam2Model, Sam2Processor
+
+def _lazy_load_sam(self):
+    if self.sam_initialized:
+        return True
+        
+    try:
+        print("🔄 Lade SAM 2 über 🤗 Transformers...")
+        
+        model_id = "facebook/sam2-hiera-tiny"  # Dieser Pfad ist korrekt
+        self.sam_processor = Sam2Processor.from_pretrained(model_id)
+        self.sam_model = Sam2Model.from_pretrained(model_id).to(self.device)
+        
+        self.sam_initialized = True
+        print("✅ SAM 2 erfolgreich geladen (via Transformers)")
+        return True
+        
+    except Exception as e:
+        print(f"❌ Fehler beim Laden von SAM 2: {e}")
+        self.sam_initialized = True
+        return False
+
+
+def _lazy_load_sam(self):
+    if self.sam_initialized:
+        return True
+        
+    try:
+        print("🔄 Lade SAM 2 über 🤗 Transformers...")
+        
+        model_id = "facebook/sam2-hiera-tiny"  # Dieser Pfad ist korrekt
+        self.sam_processor = Sam2Processor.from_pretrained(model_id)
+        self.sam_model = Sam2Model.from_pretrained(model_id).to(self.device)
+        
+        self.sam_initialized = True
+        print("✅ SAM 2 erfolgreich geladen (via Transformers)")
+        return True
+        
+    except Exception as e:
+        print(f"❌ Fehler beim Laden von SAM 2: {e}")
+        self.sam_initialized = True
+        return False
         
     def _lazy_load_sam(self):
         """Lazy Loading von SAM 2 Tiny - Optimiert für Hugging Face Spaces"""