Update app.py

fix embedding dimension, add button

app.py

@@ -13,10 +13,11 @@ model = model.merge_and_unload()
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
+st.header("Irrelevant Review Detections :sunglasses:")
 description = st.text_input("Product description")
 review = st.text_input("Review")
 
-if description and review:
+if st.button("Detect") and description and review:
     input_texts = [
         f'query: {review}',
         f'passage: {description}'
@@ -26,10 +27,13 @@ if description and review:
 
     query_embedding, doc_embedding = model(**batch_dict, return_dict=True).pooler_output
 
+    query_embedding = query_embedding.unsqueeze(0)
+    doc_embedding = doc_embedding.unsqueeze(0)
+
     similarity = torch.nn.functional.cosine_similarity(
         query_embedding, doc_embedding)
 
-    threshold = 0.7
+    threshold = 0.83
 
     if similarity > threshold:
         st.write('Relevant')