Update app.py

app.py

@@ -31,8 +31,15 @@ class SketchToRealisticFace:
         # )
 
         # Load Realistic LoRA model
-        self.pipe = StableDiffusionControlNetPipeline.from_pretrained(
-            "SG161222/Realistic_Vision_V6.0_B1_noVAE",
+        # self.pipe = StableDiffusionControlNetPipeline.from_pretrained(
+        #     "SG161222/Realistic_Vision_V6.0_B1_noVAE",
+        #     controlnet=self.controlnet,
+        #     torch_dtype=torch.float16,
+        #     safety_checker=None,
+        #     requires_safety_checker=False
+        # )
+        self.pipe = StableDiffusionControlNetPipeline.from_single_file(
+            "https://huggingface.co/NikhilJoson/Realistic_Vision_V6_B1/resolve/main/realisticVisionV60B1_v51HyperVAE.safetensors",
             controlnet=self.controlnet,
             torch_dtype=torch.float16,
             safety_checker=None,