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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -1,9 +1,13 @@
1
  import streamlit as st
2
  import torch
3
  from sentence_transformers import SentenceTransformer, util
 
 
 
4
 
5
  # Load the pre-trained SentenceTransformer model
6
  model = SentenceTransformer('all-MiniLM-L6-v2')
 
7
 
8
  # Define the backend function
9
  def mapping_code(user_input):
 
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
13
  def mapping_code(user_input):