import os
import gradio as gr

model_list = ['utrobinmv/t5_translate_en_ru_zh_large_1024_v2',
              'utrobinmv/t5_translate_en_ru_zh_large_1024',  
              'utrobinmv/t5_translate_en_ru_zh_small_1024',
              'utrobinmv/t5_translate_en_ru_zh_base_200',
              'Helsinki-NLP/opus-mt-en-ru',
              'Helsinki-NLP/opus-mt-en-zh',
              'Helsinki-NLP/opus-mt-ru-en',
              'Helsinki-NLP/opus-mt-zh-en',
              'facebook/m2m100_1.2B',
              'facebook/m2m100_418M',
              'facebook/mbart-large-50-many-to-many-mmt',
              'facebook/nllb-200-1.3B',
              'facebook/nllb-200-3.3B',
              'facebook/nllb-200-distilled-1.3B',
              'facebook/nllb-200-distilled-600M',
              'facebook/wmt19-en-ru',
              'jbochi/madlad400-3b-mt',
              'jbochi/madlad400-10b-mt',
              'google/madlad400-3b-mt',
              'google/madlad400-10b-mt',
              'joefox/mbart-large-ru-zh-ru-many-to-many-mmt']

def load_csv():
    filename = 'table.html'
    with open(filename, 'r', encoding="utf-8-sig") as f:
        return f.read()
    return ''

iface = gr.Interface(
    fn=load_csv,
    inputs=None,
    outputs="html",
    title="NTREX dataset Benchmark!",
    description="Benchmark translation to TREX dataset for ru, en, zh languages. Dataset: https://github.com/MicrosoftTranslator/NTREX",
    live=True
)

iface.launch(share=False)