Spaces:
Running
Running
from transformers import AutoTokenizer, AutoModelForSequenceClassification | |
import torch | |
import gradio as gr | |
model_name = "Tlezz324/thai-scam-detector-v1.69" | |
tokenizer = AutoTokenizer.from_pretrained(model_name) | |
model = AutoModelForSequenceClassification.from_pretrained(model_name) | |
def predict(text): | |
inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True) | |
with torch.no_grad(): | |
logits = model(**inputs).logits | |
pred = torch.argmax(logits, dim=1).item() | |
return "Scam" if pred == 1 else "Not Scam" | |
iface = gr.Interface( | |
fn=predict, | |
inputs=gr.Textbox(lines=4, placeholder="พิมพ์ข้อความที่ต้องการตรวจสอบ..."), | |
outputs="text", | |
title="Thai Scam Detector Demo", | |
description="โมเดลตรวจจับข้อความหลอกลวงภาษาไทย" | |
) | |
if __name__ == "__main__": | |
iface.launch() | |