Update app.py

app.py

@@ -48,8 +48,8 @@ encoded_sent = tokenizer.encode(
 indexed_tokens = tokenizer.convert_tokens_to_ids(tokenizer.tokenize(str(text)))#le.convert_tokens_to_ids(tkns)
 segments_ids = [0] * len(indexed_tokens)
 
-tokens_tensor = torch.tensor([indexed_tokens]).to(device)
-segments_tensors = torch.tensor([segments_ids]).to(device)
+tokens_tensor = torch.tensor([indexed_tokens])#.to(device)
+segments_tensors = torch.tensor([segments_ids])#.to(device)
 
 model = BertForSequenceClassification.from_pretrained(
     "bert-base-uncased", # Use the 12-layer BERT model, with an uncased vocab.
@@ -59,7 +59,7 @@ model = BertForSequenceClassification.from_pretrained(
     output_hidden_states = False, # Whether the model returns all hidden-states.
 )
 model.load_state_dict(torch.load("model_last_version.pt"))
-model.to(device)
+# model.to(device)
 model.eval()
 with torch.no_grad():
     logit = model(tokens_tensor,