Spaces:

qisan
/

Depressed_sentimental_analysis

Runtime error

liangc40 commited on Jan 11, 2023

Commit

1389140

•

1 Parent(s): baf29f9

Update app.py

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)