Update app.py

app.py

@@ -27,16 +27,10 @@ def get_similarity_scores(queries: list, passages: list, model, tokenizer):
     scores = (embeddings[:len(queries)] @ embeddings[len(queries):].T) * 100
     return scores.tolist()
 
-def similarity_ui(keyName, field1, field2, field3):
+def similarity_ui(keyNames:list, fields:list):
     task = 'Given a keyName, find similarity score against provided fields'
-    queries = [
-        keyName
-    ]
-    passages = [
-        field1,
-        field2,
-        field3
-    ]
+    queries = keyNames
+    passages = fields
 
     scores = get_similarity_scores(queries, passages, model, tokenizer)
     return {'Similarity Scores': scores}
@@ -48,7 +42,7 @@ model = AutoModel.from_pretrained('Salesforce/SFR-Embedding-Mistral')
 # Create Gradio Interface
 gr.Interface(
     fn=similarity_ui, 
-    inputs=["text", "text","text","text",], 
+    inputs="text", "text", 
     outputs="text",
     title="Similarity Score Calculator",
     description="Enter a Key Name and 3 Fields to find similarity scores"