PBusienei commited on
Commit
846c9b9
1 Parent(s): f482a30

Call the pipeline from transformers

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import streamlit as st
2
  import pandas as pd
3
  import altair as alt
 
4
  #!pip install -U sentence-transformers
5
  #from sentence_transformers import SentenceTransformer, util
6
  #from sentence_transformers import SentenceTransformer
@@ -52,7 +53,7 @@ def main():
52
  locations = list(df["Location Name"])
53
  # Query
54
  # Load the model
55
- model = SentenceTransformer('sentence-transformers/multi-qa-MiniLM-L6-cos-v1')
56
 
57
  query = st.text_input("Enter your query: ")
58
 
@@ -61,7 +62,7 @@ def main():
61
  #age = st.number_input("Age in Years")
62
  #Encode query and documents
63
  query_emb = model.encode(query).astype(float)
64
-
65
  #Compute dot score between query and all document embeddings
66
  scores = util.dot_score(query_emb, doc_emb.astype(float))[0].cpu().tolist()
67
 
 
1
  import streamlit as st
2
  import pandas as pd
3
  import altair as alt
4
+ from transformers import pipeline
5
  #!pip install -U sentence-transformers
6
  #from sentence_transformers import SentenceTransformer, util
7
  #from sentence_transformers import SentenceTransformer
 
53
  locations = list(df["Location Name"])
54
  # Query
55
  # Load the model
56
+ model = pipeline('sentence-transformers/multi-qa-MiniLM-L6-cos-v1')
57
 
58
  query = st.text_input("Enter your query: ")
59
 
 
62
  #age = st.number_input("Age in Years")
63
  #Encode query and documents
64
  query_emb = model.encode(query).astype(float)
65
+
66
  #Compute dot score between query and all document embeddings
67
  scores = util.dot_score(query_emb, doc_emb.astype(float))[0].cpu().tolist()
68