Spaces:

qisan
/

Depressed_sentimental_analysis

Runtime error

liangc40 commited on Jan 11, 2023

Commit

89b3965

•

1 Parent(s): baf29f9

Update app.py (#6)

- Update app.py (1389140c8515c5a56ceb4fe0085a4f05f8dcc5d8)

Co-authored-by: Chen Liang <liangc40@users.noreply.huggingface.co>

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ project = hopsworks.login(project='liangc40')
 mr = project.get_model_registry()
 model = mr.get_model("sentimental_analysis_model", version=1)
 model_dir = model.download()
-model = joblib.load(model_dir + "/sentimental_analysis_model.pkl")
 def analyse(text):
     label = model(text)

 mr = project.get_model_registry()
 model = mr.get_model("sentimental_analysis_model", version=1)
 model_dir = model.download()
+model = torch.load(model_dir + "/sentimental_analysis_model.pkl", map_location=torch.device('cpu'))
 def analyse(text):
     label = model(text)