Update app.py

app.py

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