JoBeer commited on
Commit
ef4dea0
·
verified ·
1 Parent(s): 4cc15a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -5,12 +5,15 @@ import torch
5
  from sentence_transformers.util import semantic_search
6
  import pandas as pd
7
  import requests
 
8
 
9
  #Import corpus embeddings
10
- corpus_ger = pd.read_json('corpus_embed_ger_model5.json')
11
  corpus_eng = pd.read_json('corpus_embed_eng_mboth.json')
12
  corpus_fr = pd.read_json('corpus_embed_fr_Sahajtomar.json')
13
 
 
 
14
  #Import models
15
  model_ger = SentenceTransformer('JoBeer/model5')
16
  model_eng = SentenceTransformer('mboth/distil-eng-quora-sentence')
 
5
  from sentence_transformers.util import semantic_search
6
  import pandas as pd
7
  import requests
8
+ from datasets import load_dataset
9
 
10
  #Import corpus embeddings
11
+ #corpus_ger = pd.read_json('corpus_embed_ger_model5.json')
12
  corpus_eng = pd.read_json('corpus_embed_eng_mboth.json')
13
  corpus_fr = pd.read_json('corpus_embed_fr_Sahajtomar.json')
14
 
15
+ corpus_ger = load_dataset('JoBeer/eclass_ger', use_auth_token=True)
16
+
17
  #Import models
18
  model_ger = SentenceTransformer('JoBeer/model5')
19
  model_eng = SentenceTransformer('mboth/distil-eng-quora-sentence')