Update app.py

app.py

@@ -59,6 +59,7 @@ processor = LinguisticPreprocessor()
 train_data = load_dataset('rotten_tomatoes', split='train')
 vocab, tokenizer = build_vocab(train_data)
 stop_words = set(stopwords.words('english'))
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 ############################################################# Text input