ceejaytheanalyst commited on
Commit
db10834
·
verified ·
1 Parent(s): 0e9a048

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,12 +1,14 @@
1
  import streamlit as st
2
  import torch
3
  from sentence_transformers import SentenceTransformer, util
 
4
  import pandas as pd
5
  import numpy as np
6
 
7
 
8
  # Load the pre-trained SentenceTransformer model
9
- model = SentenceTransformer('all-MiniLM-L6-v2')
 
10
  sentence_embed = pd.read_csv('Reference_file.csv')
11
 
12
  # Define the backend function
 
1
  import streamlit as st
2
  import torch
3
  from sentence_transformers import SentenceTransformer, util
4
+ from transformers import pipeline
5
  import pandas as pd
6
  import numpy as np
7
 
8
 
9
  # Load the pre-trained SentenceTransformer model
10
+ pipeline = pipeline(task="Sentence Similarity", model="all-MiniLM-L6-v2")
11
+ #model = SentenceTransformer('all-MiniLM-L6-v2')
12
  sentence_embed = pd.read_csv('Reference_file.csv')
13
 
14
  # Define the backend function