Update app.py

app.py

@@ -34,10 +34,10 @@ def inference(m1,m2):
   scores = torch.softmax(output.logits, dim=-1)
   listscore = scores.tolist()
   print(listscore)
-  b['not related'] = listscore[0]
-  b['coref'] = listscore[1]
-  b['parent'] = listscore[2]
-  b['child'] = listscore[3]
+  b['not related'] = listscore[0][0]
+  b['coref'] = listscore[0][1]
+  b['parent'] = listscore[0][2]
+  b['child'] = listscore[0][3]
   return b
 
 title = "Longformer-scico"