Spaces:

PBusienei
/

Summit_app_demo

Runtime error

PBusienei commited on Apr 11, 2022

Commit

8cc1051

1 Parent(s): 609731d

updated the model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -53,7 +53,7 @@ def main():
     locations = list(df["Location Name"])
 # Query
 # Load the model
-    model = pipeline('multi-qa-MiniLM-L6-cos-v1')#('sentence-transformers/multi-qa-MiniLM-L6-cos-v1')
     query =  st.text_input("Enter your query: ")

     locations = list(df["Location Name"])
 # Query
 # Load the model
+    model = pipeline('text-classification')#('sentence-transformers/multi-qa-MiniLM-L6-cos-v1')
     query =  st.text_input("Enter your query: ")