Spaces:
Runtime error
Runtime error
Call the pipeline from transformers
Browse files
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 =
|
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 |
|