TruthLens commited on
Commit
ca9580f
·
verified ·
1 Parent(s): 4d34795

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -10,8 +10,11 @@ app = Flask(__name__)
10
  upload_folder = os.path.join('static', 'uploads')
11
  os.makedirs(upload_folder, exist_ok=True)
12
 
13
- # Better Fake News Detection Model (DeBERTa for improved performance)
14
- news_model = pipeline("text-classification", model="yn2022/fake-news-detection")
 
 
 
15
 
16
  # Image Detection Model (CLIP-based)
17
  clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
 
10
  upload_folder = os.path.join('static', 'uploads')
11
  os.makedirs(upload_folder, exist_ok=True)
12
 
13
+ news_models = {
14
+ "mrm8488": pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-fake-news-detection"),
15
+ "google-electra": pipeline("text-classification", model="google/electra-base-discriminator"),
16
+ "bert-base": pipeline("text-classification", model="bert-base-uncased")
17
+ }
18
 
19
  # Image Detection Model (CLIP-based)
20
  clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")