Update tasks/text.py

tasks/text.py

@@ -100,12 +100,13 @@ async def evaluate_text(request: TextEvaluationRequest):
     tokenizer = AutoTokenizer.from_pretrained(BINARY_MODEL)
     print('Loaded Tokenizer')
     model = AutoModelForSequenceClassification.from_pretrained(BINARY_MODEL)
-    print('Loaded Model')
+    print(device)
     model.to(device)
     model.eval()
-
+    print('Loaded Model')
     predictions = []
     for i,text in tqdm(enumerate(test_dataset["quote"])):
+        print(i)
         with torch.no_grad():        
             tokenized_text = tokenizer(text, truncation=True, padding='max_length', return_tensors = "pt")
             inputt = {k:v.to(device) for k,v in tokenized_text.items()}
@@ -116,8 +117,6 @@ async def evaluate_text(request: TextEvaluationRequest):
             
             prediction = "0_not_relevant" if binary_prediction==0 else 1
             predictions.append(prediction)
-        if i%10:
-            print(f'iteration: {i}')
             
     gc.collect()