Ankit74990 commited on
Commit
a7bb4a3
·
verified ·
1 Parent(s): 35091d4

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -1,8 +1,17 @@
 
1
  import gradio as gr
2
  import torch
 
3
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
4
 
5
- MODEL_NAME = "Ankit74990/TruthX-DISTILBERT"
 
 
 
 
 
 
 
6
 
7
  tokenizer = None
8
  model = None
 
1
+ import os
2
  import gradio as gr
3
  import torch
4
+ from dotenv import load_dotenv
5
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
6
 
7
+ load_dotenv()
8
+
9
+ MODEL_NAME = os.getenv("MODEL_NAME", "Ankit74990/TruthX-DISTILBERT")
10
+
11
+ NEWSDATA_API_KEY = os.getenv("NEWSDATA_API_KEY", "")
12
+ NEWSAPI_API_KEY = os.getenv("NEWSAPI_API_KEY", "")
13
+ GNEWS_API_KEY = os.getenv("GNEWS_API_KEY", "")
14
+ MEDIASTACK_API_KEY = os.getenv("MEDIASTACK_API_KEY", "")
15
 
16
  tokenizer = None
17
  model = None