Update app.py

app.py

@@ -10,13 +10,15 @@ import os
 
 #Import corpus embeddings
 #corpus_ger = pd.read_json('corpus_embed_ger_model5.json')
-corpus_eng = pd.read_json('corpus_embed_eng_mboth.json')
-corpus_fr = pd.read_json('corpus_embed_fr_Sahajtomar.json')
+#corpus_eng = pd.read_json('corpus_embed_eng_mboth.json')
+#corpus_fr = pd.read_json('corpus_embed_fr_Sahajtomar.json')
 
-corpus_ger = load_dataset('JoBeer/corpus_ger', use_auth_token=str(os.environ['hf_token']))
+corpus_ger = pd.DataFrame(load_dataset('JoBeer/corpus_ger', use_auth_token=True)['train'])
+corpus_eng = pd.DataFrame(load_dataset('JoBeer/corpus_eng', use_auth_token=True)['train'])
+corpus_fr = pd.DataFrame(load_dataset('JoBeer/corpus_fr', use_auth_token=True)['train'])
 
 #Import models
-model_ger = SentenceTransformer('JoBeer/model5')
+model_ger = SentenceTransformer('JoBeer/model5') #,use_auth_token=str(os.environ['hf_token'])
 model_eng = SentenceTransformer('mboth/distil-eng-quora-sentence')
 model_fr = SentenceTransformer('Sahajtomar/french_semantic')