Update app.py

app.py

@@ -126,7 +126,7 @@ def retrieve_gradio(knowledge_base: str, query: str, topk: int):
 
     query_with_instruction = "Represent this query for retrieving relavant document: " + query
     with torch.no_grad():
-        query_rep = encode([query_with_instruction])
+        query_rep = torch.Tensor(encode([query_with_instruction]))
 
     query_md5 = hashlib.md5(query.encode()).hexdigest()