added no_grad

model.py

@@ -60,8 +60,8 @@ def classify_text(text: str) -> float:
         max_length=128,
         return_tensors='pt'
     )["input_ids"]
-
-    X = model.forward(X)[-1][0].sum(axis=0)[None, :]
+    with torch.no_grad():
+        X = model.forward(X)[-1][0].sum(axis=0)[None, :]
     return classifier(X)