Update app.py

app.py

@@ -14,7 +14,7 @@ def last_token_pool(last_hidden_states: Tensor,
         batch_size = last_hidden_states.shape[0]
         return last_hidden_states[torch.arange(batch_size, device=last_hidden_states.device), sequence_lengths]
 
-def get_similarity_scores(queries: list, passages: list, model, tokenizer):
+def get_similarity_scores(queries:list, passages:list, model, tokenizer):
     tokenizer.add_eos_token = True
 
     max_length = 4096
@@ -27,7 +27,7 @@ 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(keyNames:list, fields:list):
+def similarity_ui(keyNames:List[str], fields:List[str]):
     task = 'Given a keyName, find similarity score against provided fields'
     queries = keyNames
     passages = fields
@@ -42,7 +42,7 @@ model = AutoModel.from_pretrained('Salesforce/SFR-Embedding-Mistral')
 # Create Gradio Interface
 gr.Interface(
     fn=similarity_ui, 
-    inputs="text", "text", 
+    inputs="text", 
     outputs="text",
     title="Similarity Score Calculator",
     description="Enter a Key Name and 3 Fields to find similarity scores"