sundea commited on
Commit
eaab6d6
·
1 Parent(s): c2a3962

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -105,7 +105,8 @@ def greet(text):
105
  x = import_module('models.' + model_name)
106
 
107
  config = x.Config(dataset, embedding)
108
- device=("cuda" if torch.cuda.is_available() else "cpu")
 
109
  model = models.TextCNN.Model(config)
110
 
111
  # vocab, train_data, dev_data, test_data = build_dataset(config, args.word)
 
105
  x = import_module('models.' + model_name)
106
 
107
  config = x.Config(dataset, embedding)
108
+ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
109
+
110
  model = models.TextCNN.Model(config)
111
 
112
  # vocab, train_data, dev_data, test_data = build_dataset(config, args.word)