Update app.py
Browse files
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)
|