AcikHack-Tools / app.py
ayberkimir's picture
code refactoring
5722b62
raw
history blame
585 Bytes
import gradio as gr
from interfaces.classifier_interface import ClassifierInterface
from interfaces.masked_interface import MaskedInterface
from interfaces.embed_interface import EmbedInterface
class_interface = ClassifierInterface()
masked_interface = MaskedInterface()
embed_interface = EmbedInterface()
demo = gr.TabbedInterface(
interface_list=[class_interface(),
masked_interface(),
embed_interface()],
tab_names=[
'Multiclass Classification',
'Masked Language Modeling',
'Sentence Similarity'
]
)
demo.launch()