Andrei-Iulian SĂCELEANU commited on
Commit
c9f00c5
1 Parent(s): 6d578db

debug print

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -37,8 +37,9 @@ def ssl_predict(in_text, model_type):
37
  model = FixMatchTune(encoder_name="andrei-saceleanu/ro-offense-freematch")
38
  model.cls_head.load_weights("./checkpoints/freematch_tune")
39
 
 
40
  preds, _ = model([toks["input_ids"],toks["attention_mask"]], training=False)
41
-
42
  probs = list(preds[0].numpy())
43
 
44
  return {k:v for k, v in zip(label_names, probs)}
 
37
  model = FixMatchTune(encoder_name="andrei-saceleanu/ro-offense-freematch")
38
  model.cls_head.load_weights("./checkpoints/freematch_tune")
39
 
40
+ print(toks["input_ids"])
41
  preds, _ = model([toks["input_ids"],toks["attention_mask"]], training=False)
42
+ print(preds)
43
  probs = list(preds[0].numpy())
44
 
45
  return {k:v for k, v in zip(label_names, probs)}