wilwork commited on
Commit
0ebdae4
·
verified ·
1 Parent(s): 90c19e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name)
10
  # Function to compute relevance score
11
  def get_relevance_score(query, paragraph):
12
  inputs = tokenizer(query, paragraph, return_tensors="pt", truncation=True, padding=True)
 
13
  with torch.no_grad():
14
  scores = model(**inputs).logits.squeeze().item()
15
  return round(scores, 4)
 
10
  # Function to compute relevance score
11
  def get_relevance_score(query, paragraph):
12
  inputs = tokenizer(query, paragraph, return_tensors="pt", truncation=True, padding=True)
13
+ model.eval()
14
  with torch.no_grad():
15
  scores = model(**inputs).logits.squeeze().item()
16
  return round(scores, 4)