Spaces:

sundea
/

text-classification

Runtime error

sundea commited on Jun 12, 2023

Commit

f6555cd

1 Parent(s): 1c3d1bf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -105,7 +105,7 @@ def greet(text):
     x = import_module('models.' + model_name)
     config = x.Config(dataset, embedding)
-    device = 'cuda:0'
     model = models.TextCNN.Model(config)
     # vocab, train_data, dev_data, test_data = build_dataset(config, args.word)

     x = import_module('models.' + model_name)
     config = x.Config(dataset, embedding)
+    device = 'cpu'
     model = models.TextCNN.Model(config)
     # vocab, train_data, dev_data, test_data = build_dataset(config, args.word)