Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ logging.basicConfig(level=logging.INFO)
|
|
19 |
logger = logging.getLogger(__name__)
|
20 |
|
21 |
# Paths for saving artifacts
|
22 |
-
MODEL_DIR = "
|
23 |
UNIVERSAL_MODEL_PATH = os.path.join(MODEL_DIR, "universal_model")
|
24 |
DETECTOR_MODEL_PATH = os.path.join(MODEL_DIR, "detector_model")
|
25 |
TFIDF_PATH = os.path.join(MODEL_DIR, "tfidf_vectorizer.pkl")
|
|
|
19 |
logger = logging.getLogger(__name__)
|
20 |
|
21 |
# Paths for saving artifacts
|
22 |
+
MODEL_DIR = "/data/saved_models" # Use /data for persistent storage in Hugging Face Spaces
|
23 |
UNIVERSAL_MODEL_PATH = os.path.join(MODEL_DIR, "universal_model")
|
24 |
DETECTOR_MODEL_PATH = os.path.join(MODEL_DIR, "detector_model")
|
25 |
TFIDF_PATH = os.path.join(MODEL_DIR, "tfidf_vectorizer.pkl")
|