Update app.py

app.py

@@ -1,12 +1,13 @@
 import gradio as gr
 import spaces
 import torch
-from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, AutoModel, pipeline, logging
+from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, AutoModel, pipeline
+from transformers import logging as hflogging
 import languagecodes
 import requests, os
 import polars as pl 
 
-logging.set_verbosity_error()
+hflogging.set_verbosity_error()
 favourite_langs = {"German": "de", "Romanian": "ro", "English": "en", "-----": "-----"}
 df = pl.read_parquet("isolanguages.parquet")
 non_empty_isos = df.slice(1).filter(pl.col("ISO639-1") != "").rows()
@@ -444,18 +445,20 @@ def create_interface():
         with gr.Row():
             s_language = gr.Dropdown(choices=options, value = options[0], label="Source language", interactive=True)
             t_language = gr.Dropdown(choices=options, value = options[1], label="Target language", interactive=True)
-            swap_button = gr.Button("Swap Languages", size="md")
-            swap_button.click(fn=swap_languages, inputs=[s_language, t_language], outputs=[s_language, t_language], api_name=False, show_api=False)
+            swap_btn = gr.Button("Swap Languages", size="md")
+            swap_btn.click(fn=swap_languages, inputs=[s_language, t_language], outputs=[s_language, t_language], api_name=False, show_api=False)
 
-        model_name = gr.Dropdown(choices=models, label=f"Select a model. Default is {models[0]}.", value = models[0], interactive=True)
-        translate_button = gr.Button("Translate")
+        with gr.Row():
+            model_name = gr.Dropdown(choices=models, label=f"Select a model. Default is {models[0]}.", value = models[0], interactive=True, scale=2)
+            model_infobtn = gr.Button("Model Info")
+        translate_btn = gr.Button("Translate")
 
         translated_text = gr.Textbox(label="Translated text:", placeholder="Display field for translation", interactive=False, show_copy_button=True)
         message_text = gr.Textbox(label="Messages:", placeholder="Display field for status and error messages", interactive=False,
                                   value=f'Default translation settings: from {s_language.value} to {t_language.value} with {model_name.value}.')
         allmodels = gr.HTML(label="Model links:", value=', '.join([f'<a href="https://huggingface.co/{model}">{model}</a>' for model in models]))
-
-        translate_button.click(
+        model_info = gr.Markdown(label="Model info:", value=requests.get(f'https://huggingface.co/{model_name}/raw/main/README.md'))
+        translate_btn.click(
             fn=translate_text, 
             inputs=[input_text, s_language, t_language, model_name], 
             outputs=[translated_text, message_text]