Update deepfake_vs_real.py

deepfake_vs_real.py

@@ -7,7 +7,7 @@ from PIL import Image
 import torch
 
 # Load model and processor
-model_name = "prithivMLmods/Deepfake-vs-Real-8000"
+model_name = "prithivMLmods/Deepfake-Quality-Classifier2-SigLIP2"
 model = SiglipForImageClassification.from_pretrained(model_name)
 processor = AutoImageProcessor.from_pretrained(model_name)
 
@@ -23,7 +23,7 @@ def deepfake_classification(image):
         probs = torch.nn.functional.softmax(logits, dim=1).squeeze().tolist()
     
     labels = {
-        "0": "Deepfake", "1": "Real one"
+        "0": "Issue In Deepfake", "1": "High Quality Deepfake"
     }
     predictions = {labels[str(i)]: round(probs[i], 3) for i in range(len(probs))}
     
@@ -34,8 +34,8 @@ iface = gr.Interface(
     fn=deepfake_classification,
     inputs=gr.Image(type="numpy"),
     outputs=gr.Label(label="Prediction Scores"),
-    title="Deepfake vs. Real Image Classification",
-    description="Upload an image to determine if it's a Deepfake or a Real one."
+    title="Deepfake Quality Detection",
+    description="Upload an image to check its deepfake probability scores."
 )
 
 # Launch the app