MorenoLaQuatra commited on
Commit
ccd5ee8
1 Parent(s): f12a60c

Map location device

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ clf_pipeline = pipeline("text-classification", model=clf_model, tokenizer=clf_to
16
  reg_model_tag = "distilbert-base-multilingual-cased"
17
  reg_model_folder = "reg_model/regression_model.pt"
18
  reg_model = DualRegressionModel(model_name_or_path=reg_model_tag)
19
- reg_model.load_model(reg_model_folder)
20
 
21
 
22
  # define the function to be used for prediction
 
16
  reg_model_tag = "distilbert-base-multilingual-cased"
17
  reg_model_folder = "reg_model/regression_model.pt"
18
  reg_model = DualRegressionModel(model_name_or_path=reg_model_tag)
19
+ reg_model.load_model(reg_model_folder, map_location=torch.device('cpu'))
20
 
21
 
22
  # define the function to be used for prediction