edia_we_es / app.py
LMartinezEXEX's picture
Init commit
a779273
raw
history blame
1.24 kB
# --- Imports libs ---
import gradio as gr
import pandas as pd
# --- Imports modules ---
from modules.model_embbeding import Embedding
# --- Imports interfaces ---
from interfaces.interface_WordExplorer import interface as wordExplorer_interface
from interfaces.interface_BiasWordExplorer import interface as biasWordExplorer_interface
# --- Tool config ---
AVAILABLE_LOGS = True # [True | False]
LANGUAGE = "spanish" # [spanish | english]
EMBEDDING_SUBSET = "fasttext" # [fasttext | mini]
# --- Init classes ---
embedding = Embedding(
subset_name=EMBEDDING_SUBSET
)
labels = pd.read_json(f"language/{LANGUAGE}.json")["app"]
# --- Main App ---
INTERFACE_LIST = [
biasWordExplorer_interface(
embedding=embedding,
available_logs=AVAILABLE_LOGS,
lang=LANGUAGE),
wordExplorer_interface(
embedding=embedding,
available_logs=AVAILABLE_LOGS,
lang=LANGUAGE),
]
TAB_NAMES = [
labels["biasWordExplorer"],
labels["wordExplorer"],
]
iface = gr.TabbedInterface(
interface_list=INTERFACE_LIST,
tab_names=TAB_NAMES
)
iface.queue(concurrency_count=8)
iface.launch(debug=False)