Spaces:
Sleeping
Sleeping
aviseth commited on
Commit ·
4e491ac
1
Parent(s): 37f14b9
fix: default MODEL_NAMES to fine-tuned HF repos, fix .gitignore
Browse files- .gitignore +10 -0
- src/__pycache__/__init__.cpython-313.pyc +0 -0
- src/api/__pycache__/__init__.cpython-313.pyc +0 -0
- src/api/__pycache__/main.cpython-313.pyc +0 -0
- src/data/__pycache__/__init__.cpython-313.pyc +0 -0
- src/data/__pycache__/dataset.cpython-313.pyc +0 -0
- src/data/__pycache__/preprocessing.cpython-313.pyc +0 -0
- src/models/__pycache__/__init__.cpython-313.pyc +0 -0
- src/models/__pycache__/evaluate.cpython-313.pyc +0 -0
- src/models/__pycache__/inference.cpython-313.pyc +0 -0
- src/models/__pycache__/train.cpython-313.pyc +0 -0
- src/models/inference.py +3 -3
- src/utils/__pycache__/__init__.cpython-313.pyc +0 -0
- src/utils/__pycache__/gnews_client.cpython-313.pyc +0 -0
- src/utils/__pycache__/supabase_client.cpython-313.pyc +0 -0
.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-
|
| 31 |
-
"roberta": os.getenv("HF_REPO_ROBERTA", "roberta-
|
| 32 |
-
"xlnet": os.getenv("HF_REPO_XLNET", "xlnet-
|
| 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)
|
|
|