noranisa commited on
Commit
a6837e2
·
verified ·
1 Parent(s): d10727d

Create services/fake_news.py

Browse files
Files changed (1) hide show
  1. services/fake_news.py +34 -0
services/fake_news.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+
3
+ try:
4
+ fake_model = pipeline(
5
+ "text-classification",
6
+ model="mrm8488/bert-tiny-finetuned-fake-news-detection"
7
+ )
8
+ except:
9
+ fake_model = None
10
+
11
+
12
+ def detect_fake_news(texts):
13
+ try:
14
+ if fake_model is None:
15
+ return []
16
+
17
+ results = []
18
+
19
+ for t in texts[:20]:
20
+ res = fake_model(t[:512])[0]
21
+
22
+ label = "Fake" if "fake" in res["label"].lower() else "Real"
23
+
24
+ results.append({
25
+ "text": t,
26
+ "label": label,
27
+ "score": float(res["score"])
28
+ })
29
+
30
+ return results
31
+
32
+ except Exception as e:
33
+ print("❌ fake news error:", e)
34
+ return []