Update app.py

app.py

@@ -17,7 +17,7 @@ def get_model(base_name='intfloat/e5-large-v2'):
 # get normalized scores on input_texts, the final scores are
 # reported without queries, and the number of queries should
 # be denoted as in how_many_q
-def get_scores(model, tokenizer, input_texts, max_length=512, how_many_q=1):
+def get_scores(model, tokenizer, input_texts, max_length=512, how_many_q=1, normalize=False):
     # Tokenize the input texts
     batch_dict = tokenizer(
         input_texts,
@@ -33,9 +33,10 @@ def get_scores(model, tokenizer, input_texts, max_length=512, how_many_q=1):
     )
 
     # (Optionally) normalize embeddings
-    embeddings = F.normalize(embeddings, p=2, dim=1)
+    if normalize:
+        embeddings = F.normalize(embeddings, p=2, dim=1)
+        
     scores = (embeddings[:how_many_q] @ embeddings[how_many_q:].T) * 100
-
     return scores
 
 # get top n results out of the scores. This
@@ -107,15 +108,15 @@ tokenizer, model = get_model('intfloat/e5-large-v2')
 with gr.Blocks() as demo:
     gr.Markdown("# E5 Large V2 Demo")
     
-    q_txt = gr.Textbox(placeholder="Enter your query", info="Query")
+    q_txt = gr.Textbox(placeholder="Enter your query", label="Query")
 
-    p_txt1 = gr.Textbox(placeholder="Enter passage 1", info="Passage 1")
-    p_txt2 = gr.Textbox(placeholder="Enter passage 2", info="Passage 2")
-    p_txt3 = gr.Textbox(placeholder="Enter passage 3", info="Passage 3")
-    p_txt4 = gr.Textbox(placeholder="Enter passage 4", info="Passage 4")
-    p_txt5 = gr.Textbox(placeholder="Enter passage 5", info="Passage 5")
+    p_txt1 = gr.Textbox(placeholder="Enter passage 1", label="Passage 1")
+    p_txt2 = gr.Textbox(placeholder="Enter passage 2", label="Passage 2")
+    p_txt3 = gr.Textbox(placeholder="Enter passage 3", label="Passage 3")
+    p_txt4 = gr.Textbox(placeholder="Enter passage 4", label="Passage 4")
+    p_txt5 = gr.Textbox(placeholder="Enter passage 5", label="Passage 5")
     submit = gr.Button("Submit")
-    o_txt = gr.Textbox(placeholder="Output", lines=10, interactive=False)
+    o_txt = gr.Textbox(placeholder="Output", lines=10, interactive=False, label="Output")
 
     gr.Examples(
         [