Update app.py

app.py

@@ -20,6 +20,7 @@ svm_model = joblib.load('svm_model.joblib')
 vectorizer = joblib.load("vectorizer.joblib") # global vocabulary (used for Logistic, SVC)
 tokenizer = joblib.load("tokenizer.joblib") # used for LSTM
 
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 tokenizer1 = DistilBertTokenizer.from_pretrained("tokenizer_bert")
 model = DistilBertForSequenceClassification.from_pretrained('distilbert-base-uncased', num_labels=5)
 model.load_state_dict(torch.load("fine_tuned_bert_model1.pth", map_location=torch.device('gpu')))