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)
|