Hugging Face's logo

Spaces:
EuroPython2022
/
ToxicCommentClassification
Runtime error

App Files Community
ShAnSantosh commited on
Commit
15ce92b
·
1 Parent(s): 938f076

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -66,7 +66,6 @@ class ToxicModel(torch.nn.Module):
66
 
67
  def inference_fn(model, input_ids=None, attention_mask=None, token_type_ids=None):
68
  model.eval()
69
- print(input_ids, attention_mask, token_type_ids)
70
  input_ids = input_ids[0].to(device)
71
  attention_mask = attention_mask[0].to(device)
72
  token_type_ids = token_type_ids[0].to(device)
 
66
 
67
  def inference_fn(model, input_ids=None, attention_mask=None, token_type_ids=None):
68
  model.eval()
 
69
  input_ids = input_ids[0].to(device)
70
  attention_mask = attention_mask[0].to(device)
71
  token_type_ids = token_type_ids[0].to(device)