Spaces:

Gaurav-Mhatre
/

TrueFrame

Sleeping

Gaurav-Mhatre commited on Feb 22

Commit

cef3f4d

verified ·

1 Parent(s): df25483

Update image_detect.py

Files changed (1) hide show

image_detect.py CHANGED Viewed

@@ -6,7 +6,7 @@ class ImageDeepfakeDetector:
     def __init__(self, local_model_path="./local_clip_model"):
         print("⚡ Loading Image AI Model...")
         try:
-            self.model = CLIPModel.from_pretrained(local_model_path)
             self.processor = CLIPProcessor.from_pretrained(local_model_path)
             print("✅ Image Model Ready.")
         except Exception as e:

     def __init__(self, local_model_path="./local_clip_model"):
         print("⚡ Loading Image AI Model...")
         try:
+            self.model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
             self.processor = CLIPProcessor.from_pretrained(local_model_path)
             print("✅ Image Model Ready.")
         except Exception as e: