JoBeer commited on
Commit
72d7a7f
·
1 Parent(s): bbeafaf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,12 +3,13 @@ import sentence_transformers
3
  from sentence_transformers import SentenceTransformer
4
  import torch
5
  from sentence_transformers.util import semantic_search
6
-
7
  import pandas as pd
8
 
9
  model = SentenceTransformer('JoBeer/all-mpnet-base-v2-eclass')
10
 
11
- corpus = pd.read_excel('corpus.xlsx')
 
12
 
13
  def predict(name, description):
14
  text = 'Description: '+ description + '; Name: ' + name
 
3
  from sentence_transformers import SentenceTransformer
4
  import torch
5
  from sentence_transformers.util import semantic_search
6
+ import os
7
  import pandas as pd
8
 
9
  model = SentenceTransformer('JoBeer/all-mpnet-base-v2-eclass')
10
 
11
+ #corpus = pd.read_excel('corpus.xlsx')
12
+ corpus = pd.read_json('corpus.jsonl', lines = True, encoding = 'utf-8')
13
 
14
  def predict(name, description):
15
  text = 'Description: '+ description + '; Name: ' + name