Update app.py

app.py

@@ -1,6 +1,4 @@
 import gradio as gr
-import gradio.inputs
-import gradio.outputs
 from transformers.pipelines import pipeline
 
 model_names = ["IT5 Small", "IT5 Base", "IT5 Large", "mT5 Small", "mT5 Base", "IT5 Small Efficient"]
@@ -11,7 +9,7 @@ task_values = ['headline-generation', 'question-answering', 'question-generation
 
 full_ids = {
     task_id: {
-        model_id: f"it5/{model_id}-{task_id}"
+        model_id: f"gsarti/{model_id}-{task_id}"
         for model_id in model_values
     }
     for task_id in task_values
@@ -84,13 +82,13 @@ def get_outputs(text, model_id_idx, task_id_idx):
 iface = gr.Interface(
     fn=get_outputs,
     inputs=[
-        gradio.inputs.Textbox(label="Text", lines=20, placeholder="Enter a text here..."),
-        gradio.inputs.Dropdown(label="Model", choices=model_names, type="index"),
-        gradio.inputs.Dropdown(label="Task", choices=task_names, type="index")
+        gr.Textbox(label="Text", lines=20, placeholder="Enter a text here..."),
+        gr.Dropdown(label="Model", choices=model_names, type="index"),
+        gr.Dropdown(label="Task", choices=task_names, type="index")
     ],
     outputs=[
-        gradio.outputs.Textbox(label="Generated text:"),
-        gradio.outputs.HTML(label="Model used:"),
+        gr.Textbox(label="Generated text:"),
+        gr.HTML(label="Model used:"),
     ],
     examples = examples,
     title="IT5 Sequence-to-sequence Demo 🇮🇹",