Update app.py

app.py

@@ -8,7 +8,7 @@ def save_embeddings(sentences, filename):
     torch.save(embeddings, filename)
 
 def load_embeddings(filename):
-    return torch.load(filename)
+    return torch.load(filename, map_location=torch.device('cpu'))
 
 def preprocess_model_descriptions(file_path):
     encodings = ['utf-8', 'latin-1', 'utf-16']
@@ -44,13 +44,13 @@ embeddings = load_embeddings('embeddings_hf_spaces_descriptions.pt')
 
 with gr.Blocks() as demo:
     input = gr.Textbox(label="Enter your query")
-    button = gr.Button(label="Search")
+    num_results = gr.Slider(10, 100, value=10, label="Number of results")
     
     df_output = gr.Dataframe(label="Similarity Results", wrap=True)
     
     def search(query):
         query_embedding = model.encode([query], convert_to_tensor=True)
-        return perform_similarity_search(query_embedding, embeddings, model_ids, descriptions)
+        return perform_similarity_search(query_embedding, embeddings, model_ids, descriptions, top_k=num_results)
     
     input.submit(search, inputs=input, outputs=df_output)
     button.click(search, inputs=input, outputs=df_output)