jonathanjordan21 commited on
Commit
93ab5aa
·
1 Parent(s): 96769b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@ model = load_model()
15
 
16
  @st.cache_data(persist='disk')
17
  def embed_data(data):
18
- corpus_embeddings = model.encode(data)
19
-
20
 
21
  def process_data(df, desc, message, embed=False):
22
  data = [
 
15
 
16
  @st.cache_data(persist='disk')
17
  def embed_data(data):
18
+ return model.encode(data)
19
+
20
 
21
  def process_data(df, desc, message, embed=False):
22
  data = [