Muhammad541 commited on
Commit
d06a70a
·
verified ·
1 Parent(s): f3fc06a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "./saved_models"
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")