aaa

app.py

@@ -28,7 +28,7 @@ def correlation_score(table, review):
     # Compute the correlation score
     args = ((table, review))
     inputs = bert_tokenizer(*args, padding=True, max_length=128, truncation=True, return_tensors="pt")
-    logits = model(**inputs).logits
+    logits = bert_model(**inputs).logits
     probs = logits.softmax(dim=-1)
     return {
         "correlation": probs[:, 1].item()