Sleepyp00 commited on
Commit
c448a5b
·
1 Parent(s): cb1b925

use fine tuned model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,9 +8,9 @@ project = hopsworks.login()
8
  fs = project.get_feature_store()
9
 
10
  mr = project.get_model_registry()
11
- classifier = mr.get_model("base_classifier", version = 1)
12
  model_dir = classifier.download()
13
- classifier = joblib.load(model_dir + "/base_classifier.pkl")
14
 
15
  embedding_model = mr.get_model("news_embedding", version = 1)
16
  model_dir = embedding_model.download()
 
8
  fs = project.get_feature_store()
9
 
10
  mr = project.get_model_registry()
11
+ classifier = mr.get_model("finetuned_classifier", version = 1)
12
  model_dir = classifier.download()
13
+ classifier = joblib.load(model_dir + "/finetuned_classifier.pkl")
14
 
15
  embedding_model = mr.get_model("news_embedding", version = 1)
16
  model_dir = embedding_model.download()