Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
•
a41e9e6
1
Parent(s):
abbefff
Update app.py
Browse files
app.py
CHANGED
|
@@ -33,6 +33,7 @@ def inference(m1,m2):
|
|
| 33 |
|
| 34 |
scores = torch.softmax(output.logits, dim=-1)
|
| 35 |
listscore = scores.tolist()
|
|
|
|
| 36 |
b['not related'] = listscore[0]
|
| 37 |
b['coref'] = listscore[1]
|
| 38 |
b['parent'] = listscore[2]
|
|
|
|
| 33 |
|
| 34 |
scores = torch.softmax(output.logits, dim=-1)
|
| 35 |
listscore = scores.tolist()
|
| 36 |
+
print(listscore)
|
| 37 |
b['not related'] = listscore[0]
|
| 38 |
b['coref'] = listscore[1]
|
| 39 |
b['parent'] = listscore[2]
|