Update app.py

app.py

@@ -1,3 +1,101 @@
+Vestiaire Autocomplete
+1
+2
+3
+4
+5
+6
+7
+8
+9
+10
+11
+12
+13
+14
+15
+16
+17
+18
+19
+20
+21
+22
+23
+24
+25
+26
+27
+28
+29
+30
+31
+32
+33
+34
+35
+36
+37
+38
+39
+40
+41
+42
+43
+44
+45
+46
+47
+48
+49
+50
+51
+52
+53
+54
+55
+56
+57
+58
+59
+60
+61
+62
+63
+64
+65
+66
+67
+68
+69
+70
+71
+72
+73
+74
+75
+76
+77
+78
+79
+80
+81
+82
+83
+84
+85
+86
+87
+88
+89
+90
+91
+92
+93
+94
+95
+96
+97
 import gradio as gr
 import kagglehub
 from sentence_transformers import SentenceTransformer, util
@@ -38,10 +136,11 @@ model = SentenceTransformer(model_name)
 # Function to find synonyms dynamically with fallback
 def find_synonym(word, top_n=1):
     query_embedding = model.encode(word, convert_to_tensor=True)
-    results = util.semantic_search(query_embedding, model.encode(designer_data + category_data, convert_to_tensor=True), top_k=top_n)
+    combined_data = designer_data + category_data
+    results = util.semantic_search(query_embedding, model.encode(combined_data, convert_to_tensor=True), top_k=top_n)
     # Check if results exist
     if results and len(results[0]) > 0:
-        return [designer_data + category_data[result['corpus_id']] for result in results[0] if result['score'] > 0.6]
+        return [combined_data[result['corpus_id']] for result in results[0] if result['score'] > 0.6]
     return []  # Return an empty list if no results
 
 # Function to correct spellings
@@ -94,3 +193,4 @@ with gr.Blocks() as demo:
     )
 
 demo.launch(share=True)
+