atlas / config,py
ANISA09's picture
Update config,py
41cfad9 verified
raw
history blame contribute delete
975 Bytes
# config.py
import os
SERPAPI_KEY = os.getenv("SERPAPI_KEY")
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
SAFE_BROWSING_API_KEY = os.getenv("SAFE_BROWSING_API_KEY")
HF_SUMMARIZER = os.getenv("HF_SUMMARIZER", "sshleifer/distilbart-cnn-12-6")
HF_ZERO_SHOT = os.getenv("HF_ZERO_SHOT", "facebook/bart-large-mnli")
HF_IMAGE_CAPTION = os.getenv("HF_IMAGE_CAPTION", "nlpconnect/vit-gpt2-image-captioning")
HF_IMAGE_CLASSIFIER = os.getenv("HF_IMAGE_CLASSIFIER", "google/vit-base-patch16-224")
HF_DEEPFAKE_MODEL = os.getenv("HF_DEEPFAKE_MODEL", "prithivMLmods/deepfake-detector-model-v1")
GENAI_MODEL = os.getenv("GENAI_MODEL", "gemini-2.5-flash")
CANDIDATE_LABELS = ["True", "False", "Misleading", "Unclear", "Opinionated", "Unsupported"]
SOURCE_TRUST = {
"reuters.com": 0.95,
"apnews.com": 0.95,
"bbc.com": 0.93,
"theguardian.com": 0.9,
"nytimes.com": 0.9,
"washingtonpost.com": 0.9,
"aljazeera.com": 0.88,
"cnn.com": 0.88,
"msn.com": 0.8,
}