Spaces:

Daeyongkwon98
/

User_Intents_and_Musical_Attributes_Classifier

Sleeping

App Files Files Community

Daeyongkwon98 commited on Oct 23

Commit

4811d2a

•

1 Parent(s): dbe00b2

load model to cpu

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def classify(sentence):
 		num_intents = num_intents_dict[data_type]
 		model = DistilBERTClass(num_intents)
-		model.load_state_dict(torch.load(f"./models/{data_type}_finetune_model.pth"))
 		model.to(device)
 		# Tokenize the input sentence

 		num_intents = num_intents_dict[data_type]
 		model = DistilBERTClass(num_intents)
+		model.load_state_dict(torch.load(f"./models/{data_type}_finetune_model.pth", map_location=torch.device('cpu')))
 		model.to(device)
 		# Tokenize the input sentence