jeevitha-app commited on
Commit
1ef9d51
·
verified ·
1 Parent(s): 6e629f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -15,10 +15,10 @@ import math
15
  import gradio as gr
16
 
17
  # --------- Load models (replace filenames if you used different names) ----------
18
- ENG_MODEL_PATH = "models/english_lr_model.pkl"
19
- ENG_VEC_PATH = "models/english_vectorizer.pkl"
20
- PER_MODEL_PATH = "models/persian_lr_model.pkl"
21
- PER_VEC_PATH = "models/persian_vectorizer.pkl"
22
 
23
  eng_model = joblib.load(ENG_MODEL_PATH)
24
  eng_vectorizer = joblib.load(ENG_VEC_PATH)
 
15
  import gradio as gr
16
 
17
  # --------- Load models (replace filenames if you used different names) ----------
18
+ ENG_MODEL_PATH = "best_model.pkl"
19
+ ENG_VEC_PATH = "tfidf_vectorizer.pkl"
20
+ PER_MODEL_PATH = "logistic_regression.pkl"
21
+ PER_VEC_PATH = "tfidf_vectorizer_persian.pkl"
22
 
23
  eng_model = joblib.load(ENG_MODEL_PATH)
24
  eng_vectorizer = joblib.load(ENG_VEC_PATH)