PBusienei commited on
Commit
8cc1051
1 Parent(s): 609731d

updated the model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ def main():
53
  locations = list(df["Location Name"])
54
  # Query
55
  # Load the model
56
- model = pipeline('multi-qa-MiniLM-L6-cos-v1')#('sentence-transformers/multi-qa-MiniLM-L6-cos-v1')
57
 
58
  query = st.text_input("Enter your query: ")
59
 
 
53
  locations = list(df["Location Name"])
54
  # Query
55
  # Load the model
56
+ model = pipeline('text-classification')#('sentence-transformers/multi-qa-MiniLM-L6-cos-v1')
57
 
58
  query = st.text_input("Enter your query: ")
59