EuropeanParliament
/

eurovoc_eu

Text Classification

Model card Files Files and versions

scampion commited on Nov 17, 2023

Commit

988762a

·

1 Parent(s): 2e88644

Update handler.py

Files changed (1) hide show

handler.py +1 -0

handler.py CHANGED Viewed

@@ -71,5 +71,6 @@ class EndpointHandler:
             return_tensors='pt')
         _, prediction = self.model(item["input_ids"], item["attention_mask"])
         prediction = prediction.cpu().detach().numpy()
         return prediction

             return_tensors='pt')
         _, prediction = self.model(item["input_ids"], item["attention_mask"])
         prediction = prediction.cpu().detach().numpy()
+        print(text, prediction)
         return prediction