kapllan commited on
Commit
b4e33fb
·
verified ·
1 Parent(s): cea575d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -10,15 +10,13 @@ import gradio as gr
10
  import joblib
11
  import omikuji
12
  from huggingface_hub import snapshot_download
13
- from parltopic.utils.helper import get_main_config
14
  from prepare_everything import download_model
15
 
16
- config = get_main_config()
17
 
18
  download_model(
19
  "https://dl.fbaipublicfiles.com/fasttext/supervised-models/lid.176.bin",
20
- Path(config["paths"]["resources"]) / "lid.176.bin",
21
- )
22
 
23
  # Download the model files from Hugging Face
24
 
@@ -35,14 +33,14 @@ for repo_id in model_names:
35
 
36
  lang_model = fasttext.load_model("lid.176.bin")
37
 
38
- with open(Path(config["paths"]["datasets"]) / "label2id.json", "r") as f:
39
  label2id = js.load(f)
40
 
41
  id2label = {}
42
 
43
  for key, value in label2id.items():
44
  id2label[str(value)] = key
45
- with open(Path(config["paths"]["resources"]) / "topics_hierarchy.json", "r") as f:
46
  topics_hierarchy = js.load(f)
47
 
48
 
 
10
  import joblib
11
  import omikuji
12
  from huggingface_hub import snapshot_download
 
13
  from prepare_everything import download_model
14
 
15
+
16
 
17
  download_model(
18
  "https://dl.fbaipublicfiles.com/fasttext/supervised-models/lid.176.bin",
19
+ Path("lid.176.bin")
 
20
 
21
  # Download the model files from Hugging Face
22
 
 
33
 
34
  lang_model = fasttext.load_model("lid.176.bin")
35
 
36
+ with open(Path("label2id.json"), "r") as f:
37
  label2id = js.load(f)
38
 
39
  id2label = {}
40
 
41
  for key, value in label2id.items():
42
  id2label[str(value)] = key
43
+ with open(Path("topics_hierarchy.json"), "r") as f:
44
  topics_hierarchy = js.load(f)
45
 
46