message

app.py

@@ -40,13 +40,14 @@ def get_score(abstract):
         print(logits)
 
 
-    #outputs = model(inputs['input_ids'], attention_mask=inputs['attention_mask'])
-    #scores = torch.softmax(outputs.logits, dim=1)[0]
-
-    #probs = F.softmax(scores, dim=0)
-    #accept_prob = probs[0].item()
-
-    #print(accept_prob)
+    outputs = model(inputs['input_ids'], attention_mask=inputs['attention_mask'])
+    scores = torch.softmax(outputs.logits, dim=1)[0]
+    print(scores)
+    probs = F.softmax(scores, dim=0)
+    print(probs)
+    accept_prob = probs[0].item()
+
+    print(accept_prob)
 
     #return #accept_prob