Update app.py

app.py

@@ -63,9 +63,10 @@ class ToxicModel(torch.nn.Module):
         logits = (logits1 + logits2 + logits3 + logits4 + logits5) / 5
         return logits
  
+ 
 def inference_fn(model, input_ids=None, attention_mask=None, token_type_ids=None):  
     model.eval()
-    
+    print(input_ids, attention_mask, token_type_ids)
     input_ids = input_ids.to(device)  
     attention_mask = attention_mask.to(device)  
     token_type_ids = token_type_ids.to(device)