Spaces:
Paused
Paused
Andrei-Iulian SĂCELEANU
commited on
Commit
•
c9f00c5
1
Parent(s):
6d578db
debug print
Browse files
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)}
|