Spaces:

sundea
/

text-classification

Runtime error

sundea commited on Jun 12, 2023

Commit

a745fd3

1 Parent(s): eaab6d6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -110,7 +110,7 @@ def greet(text):
     model = models.TextCNN.Model(config)
     # vocab, train_data, dev_data, test_data = build_dataset(config, args.word)
-    model.load_state_dict(torch.load('THUCNews/saved_dict/TextCNN.ckpt'))
     model.to(device)
     model.eval()

     model = models.TextCNN.Model(config)
     # vocab, train_data, dev_data, test_data = build_dataset(config, args.word)
+    model.load_state_dict(torch.load('THUCNews/saved_dict/TextCNN.ckpt', map_location=torch.device('cpu')))
     model.to(device)
     model.eval()