cools commited on
Commit
3fef0e6
·
1 Parent(s): bcc614b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -31,12 +31,14 @@ st.divider()
31
  def search_index(query):
32
  index = pinecone.Index("scotus")
33
  vector = model.encode([query]).tolist()
34
- st.write(index.query(vector=vector,top_k=10,include_values=False,include_metadata=True))
35
- return
36
 
37
 
38
  search = st.text_input('Search', placeholder='First Amendment cannot protect libel')
39
- search_index(query=search)
 
 
40
 
41
 
42
 
 
31
  def search_index(query):
32
  index = pinecone.Index("scotus")
33
  vector = model.encode([query]).tolist()
34
+ return index.query(vector=vector,top_k=10,include_values=False,include_metadata=True).tolist()
35
+
36
 
37
 
38
  search = st.text_input('Search', placeholder='First Amendment cannot protect libel')
39
+ if search != "":
40
+ res = search_index(query=search)
41
+ st.write(res)
42
 
43
 
44