Update app.py

app.py

@@ -67,7 +67,8 @@ class ABSA(nn.Module):
 
 
 model = ABSA(vocab_size=len(token_2_id.keys()), num_labels=3)
-model = model.load_state_dict(torch.load('model_weights.pth'))
+model.load_state_dict(torch.load('model_weights.pth'))
+model.eval()
 
 print("Model loaded successfully")
 app = FastAPI()