Spaces:

kdevoe
/

testspace

Running

kdevoe commited on Jan 13

Commit

e6560a9

•

1 Parent(s): 6a951be

Update inference.py

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -29,6 +29,7 @@ def inference(input_text):
     model.eval()
     input = tokenizer(input_text)
     input_ids = torch.Tensor(input['input_ids'])
     mask = torch.Tensor(input['attention_mask'])
     output = model(input_ids, mask)

     model.eval()
     input = tokenizer(input_text)
     input_ids = torch.Tensor(input['input_ids'])
+    print(input_ids)
     mask = torch.Tensor(input['attention_mask'])
     output = model(input_ids, mask)