Update app.py

app.py

@@ -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[str], fields:list[str]):
+def similarity_ui(keyNames, fields):
     task = 'Given a keyName, find similarity score against provided fields'
     queries = keyNames
     passages = fields
@@ -42,8 +42,8 @@ model = AutoModel.from_pretrained('Salesforce/SFR-Embedding-Mistral')
 # Create Gradio Interface
 gr.Interface(
     fn=similarity_ui, 
-    inputs=["str", "str"],
-    outputs="text",
+    inputs=[gr.Textbox(), gr.Textbox()],
+    outputs=gr.Textbox(),,
     title="Similarity Score Calculator",
     description="Enter a Key Name and 3 Fields to find similarity scores"
 ).launch()