Spaces:
Runtime error
Runtime error
MorenoLaQuatra
commited on
Commit
•
ccd5ee8
1
Parent(s):
f12a60c
Map location device
Browse files
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
|