File size: 504 Bytes
cc06c26
 
 
 
 
6e8fa3a
 
 
 
 
 
 
cc06c26
6e8fa3a
cc06c26
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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=predict,
    inputs="text",
    outputs="label",
    title="Fake Review Detector"
).launch()