Spaces:
Running
Running
Update app.py
Browse files
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')
|