Update app.py

app.py

@@ -428,18 +428,20 @@ def load_txt2img(model_id):
             "safety_checker": None,
             "requires_safety_checker": False,
             "add_watermarker": False,
-            "allow_pickle": True,
             "cache_dir": "/tmp/models",    # Für Hugging Face Spaces
-            "use_safetensors": False     # Sichereres Format
         }
 
         # Die Modelle haben unterscgiedliche Gewichtsformate. Safetensors neu und schneller Zugriff!  
-        if model_id in SAFETENSORS_MODELS:
-            model_params["use_safetensors"] = True
-            print(f"ℹ️ Verwende safetensors für {model_id}")
-        else:
+        if model_id == "SG161222/Realistic_Vision_V6.0_B1_noVAE":
+            model_params["allow_pickle"] = False  # WICHTIG für PyTorch 2.6
             model_params["use_safetensors"] = False
-            print(f"ℹ️ Verwende .bin weights für {model_id}")
+            print("⚠️ Realistic Vision Modell - Nutzt .bin-Dateien.")
+        else:
+            model_params["allow_pickle"] = True
+            model_params["use_safetensors"] = True
+            model_params["variant"] = "fp16"
+            print("✅ Verwende SafeTensors für sicheres Laden.")
+
         
         if config.get("supports_fp16", False) and torch_dtype == torch.float16:
             model_params["variant"] = "fp16"