Spaces:
Sleeping
Sleeping
Ezhil
commited on
Commit
·
0d5ed90
1
Parent(s):
1864df7
modified code
Browse files
__pycache__/model.cpython-310.pyc
ADDED
|
Binary file (2.08 kB). View file
|
|
|
model/train_model.py
CHANGED
|
@@ -1,3 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
| 1 |
from model import train_sms_classifier
|
| 2 |
|
| 3 |
# Train the model and save it
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
# Set the Hugging Face cache directory to a writable location
|
| 3 |
+
os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface_cache"
|
| 4 |
from model import train_sms_classifier
|
| 5 |
|
| 6 |
# Train the model and save it
|