JoBeer commited on
Commit
5dd2878
·
verified ·
1 Parent(s): 955e6e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -8,18 +8,13 @@ import requests
8
  from datasets import load_dataset
9
  import os
10
 
11
- #Import corpus embeddings
12
- #corpus_ger = pd.read_json('corpus_embed_ger_model5.json')
13
- #corpus_eng = pd.read_json('corpus_embed_eng_mboth.json')
14
- #corpus_fr = pd.read_json('corpus_embed_fr_Sahajtomar.json')
15
-
16
  #Import corpus embeddings
17
  corpus_ger = pd.DataFrame(load_dataset('JoBeer/corpus_ger', use_auth_token=str(os.environ['hf_token']))['train'])
18
  corpus_eng = pd.DataFrame(load_dataset('JoBeer/corpus_eng', use_auth_token=str(os.environ['hf_token']))['train'])
19
  corpus_fr = pd.DataFrame(load_dataset('JoBeer/corpus_fr', use_auth_token=str(os.environ['hf_token']))['train'])
20
 
21
  #Import models
22
- model_ger = SentenceTransformer('JoBeer/model5') #,use_auth_token=str(os.environ['hf_token'])
23
  model_eng = SentenceTransformer('mboth/distil-eng-quora-sentence')
24
  model_fr = SentenceTransformer('Sahajtomar/french_semantic')
25
 
 
8
  from datasets import load_dataset
9
  import os
10
 
 
 
 
 
 
11
  #Import corpus embeddings
12
  corpus_ger = pd.DataFrame(load_dataset('JoBeer/corpus_ger', use_auth_token=str(os.environ['hf_token']))['train'])
13
  corpus_eng = pd.DataFrame(load_dataset('JoBeer/corpus_eng', use_auth_token=str(os.environ['hf_token']))['train'])
14
  corpus_fr = pd.DataFrame(load_dataset('JoBeer/corpus_fr', use_auth_token=str(os.environ['hf_token']))['train'])
15
 
16
  #Import models
17
+ model_ger = SentenceTransformer('JoBeer/model5', use_auth_token=str(os.environ['hf_token']))
18
  model_eng = SentenceTransformer('mboth/distil-eng-quora-sentence')
19
  model_fr = SentenceTransformer('Sahajtomar/french_semantic')
20