AIteen commited on
Commit
3ee37f1
·
verified ·
1 Parent(s): 3687f59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def similarity_ui(keyNames, fields):
37
  passages = fields.split(',')
38
 
39
  scores = get_similarity_scores(queries, passages, model, tokenizer)
40
- return {"Similarity Scores": scores}
41
 
42
  # Load model and tokenizer
43
  tokenizer = AutoTokenizer.from_pretrained('Salesforce/SFR-Embedding-Mistral')
 
37
  passages = fields.split(',')
38
 
39
  scores = get_similarity_scores(queries, passages, model, tokenizer)
40
+ return scores
41
 
42
  # Load model and tokenizer
43
  tokenizer = AutoTokenizer.from_pretrained('Salesforce/SFR-Embedding-Mistral')