qisan commited on
Commit
a3ea6a6
1 Parent(s): 230b305

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -4,16 +4,16 @@ from transformers import pipeline
4
 
5
  import hopsworks
6
  import joblib
7
- pipe = pipeline(model="liangc40/sentimental_analysis")
8
 
9
- #project = hopsworks.login(project='liangc40')
10
- #fs = project.get_feature_store()
11
 
12
 
13
- #mr = project.get_model_registry()
14
- #model = mr.get_model("sentimental_analysis_model", version=1)
15
- #model_dir = model.download()
16
- #model = joblib.load(model_dir + "/sentimental_analysis_model.pkl")
17
 
18
  def analyse(text):
19
  label = pipe(text)
 
4
 
5
  import hopsworks
6
  import joblib
7
+ #pipe = pipeline(model="liangc40/sentimental_analysis")
8
 
9
+ project = hopsworks.login(project='liangc40')
10
+ fs = project.get_feature_store()
11
 
12
 
13
+ mr = project.get_model_registry()
14
+ model = mr.get_model("sentimental_analysis_model", version=3)
15
+ model_dir = model.download()
16
+ model = joblib.load(model_dir + "/sentimental_analysis_model.pkl")
17
 
18
  def analyse(text):
19
  label = pipe(text)