test

app.py

@@ -116,7 +116,7 @@ if option == 'MILESTONE 3':
     X_train = pre_populated_tweets[0]
     batch = tokenizers[0](X_train, padding=True, truncation=True, max_length=512, return_tensors="pt")
 
-     with torch.no_grad():
+    with torch.no_grad():
         outputs = models[0](**batch, labels=torch.tensor([1, 0]))
         predictions = F.softmax(outputs.logits, dim=1)
         labels = torch.argmax(predictions, dim=1)