jeevitha-app commited on
Commit
8c4967a
·
verified ·
1 Parent(s): 5cf7d88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -9
app.py CHANGED
@@ -11,17 +11,13 @@ import os
11
  # -----------------------------------------------------
12
  # ✅ Load Models and Vectorizers
13
  # -----------------------------------------------------
14
- MODEL_DIR = "models"
15
 
16
- eng_model_path = os.path.join(MODEL_DIR, "logistic_regression_english.pkl")
17
- eng_vectorizer_path = os.path.join(MODEL_DIR, "tfidf_vectorizer_english.pkl")
18
- per_model_path = os.path.join(MODEL_DIR, "logistic_regression_persian.pkl")
19
- per_vectorizer_path = os.path.join(MODEL_DIR, "tfidf_vectorizer_persian.pkl")
 
20
 
21
- eng_model = joblib.load(eng_model_path)
22
- eng_vectorizer = joblib.load(eng_vectorizer_path)
23
- per_model = joblib.load(per_model_path)
24
- per_vectorizer = joblib.load(per_vectorizer_path)
25
 
26
  # -----------------------------------------------------
27
  # ✅ Label mapping
 
11
  # -----------------------------------------------------
12
  # ✅ Load Models and Vectorizers
13
  # -----------------------------------------------------
 
14
 
15
+ english_model = joblib.load("logistic_regression_english.pkl")
16
+ english_vectorizer = joblib.load("tfidf_vectorizer_english.pkl")
17
+
18
+ persian_model = joblib.load("logistic_regression_persian.pkl")
19
+ persian_vectorizer = joblib.load("tfidf_vectorizer_persian.pkl")
20
 
 
 
 
 
21
 
22
  # -----------------------------------------------------
23
  # ✅ Label mapping