Spaces:
Runtime error
Runtime error
use fine tuned model
Browse files
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("
|
| 12 |
model_dir = classifier.download()
|
| 13 |
-
classifier = joblib.load(model_dir + "/
|
| 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()
|