aviseth commited on
Commit
4e491ac
·
1 Parent(s): 37f14b9

fix: default MODEL_NAMES to fine-tuned HF repos, fix .gitignore

Browse files
.gitignore ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ __pycache__/
2
+ *.pyc
3
+ *.pyo
4
+ .pytest_cache/
5
+ .hypothesis/
6
+ *.egg-info/
7
+ dist/
8
+ build/
9
+ .env
10
+ models/
src/__pycache__/__init__.cpython-313.pyc DELETED
Binary file (192 Bytes)
 
src/api/__pycache__/__init__.cpython-313.pyc DELETED
Binary file (196 Bytes)
 
src/api/__pycache__/main.cpython-313.pyc DELETED
Binary file (17 kB)
 
src/data/__pycache__/__init__.cpython-313.pyc DELETED
Binary file (197 Bytes)
 
src/data/__pycache__/dataset.cpython-313.pyc DELETED
Binary file (5.85 kB)
 
src/data/__pycache__/preprocessing.cpython-313.pyc DELETED
Binary file (1.28 kB)
 
src/models/__pycache__/__init__.cpython-313.pyc DELETED
Binary file (502 Bytes)
 
src/models/__pycache__/evaluate.cpython-313.pyc DELETED
Binary file (2.53 kB)
 
src/models/__pycache__/inference.cpython-313.pyc DELETED
Binary file (18.9 kB)
 
src/models/__pycache__/train.cpython-313.pyc DELETED
Binary file (8.63 kB)
 
src/models/inference.py CHANGED
@@ -27,9 +27,9 @@ MODELS_DIR = PROJECT_ROOT / "models"
27
 
28
  # Override with HF Hub repo IDs via env vars, e.g. HF_REPO_DISTILBERT=your-username/distilbert-fakenews
29
  MODEL_NAMES = {
30
- "distilbert": os.getenv("HF_REPO_DISTILBERT", "distilbert-base-uncased"),
31
- "roberta": os.getenv("HF_REPO_ROBERTA", "roberta-base"),
32
- "xlnet": os.getenv("HF_REPO_XLNET", "xlnet-base-cased"),
33
  }
34
 
35
 
 
27
 
28
  # Override with HF Hub repo IDs via env vars, e.g. HF_REPO_DISTILBERT=your-username/distilbert-fakenews
29
  MODEL_NAMES = {
30
+ "distilbert": os.getenv("HF_REPO_DISTILBERT", "aviseth/distilbert-fakenews"),
31
+ "roberta": os.getenv("HF_REPO_ROBERTA", "aviseth/roberta-fakenews"),
32
+ "xlnet": os.getenv("HF_REPO_XLNET", "aviseth/xlnet-fakenews"),
33
  }
34
 
35
 
src/utils/__pycache__/__init__.cpython-313.pyc DELETED
Binary file (198 Bytes)
 
src/utils/__pycache__/gnews_client.cpython-313.pyc DELETED
Binary file (6.04 kB)
 
src/utils/__pycache__/supabase_client.cpython-313.pyc DELETED
Binary file (4.75 kB)