ceejaytheanalyst commited on
Commit
18f7759
·
verified ·
1 Parent(s): 98b1b0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,14 +1,14 @@
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
 
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