Update app.py

app.py

@@ -1,11 +1,17 @@
-# app.py
 import gradio as gr
 from transformers import pipeline
 
 detector = pipeline("text-classification", model="debojit01/fake-review-detector")
 
+def predict(text):
+    result = detector(text)[0]
+    return {
+        "Fake": result["score"] if result["label"] == "LABEL_0" else 1 - result["score"],
+        "Genuine": 1 - result["score"] if result["label"] == "LABEL_0" else result["score"]
+    }
+
 gr.Interface(
-    fn=lambda x: detector(x)[0],
+    fn=predict,
     inputs="text",
     outputs="label",
     title="Fake Review Detector"