Update app.py

app.py

@@ -1,7 +1,7 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
 import spacy
 import torch
 import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
 nlp = spacy.load('es_core_news_sm')
 
@@ -45,5 +45,10 @@ outputs_process = gr.outputs.Textbox(label="Processed Sentence")
 generate_interface = gr.Interface(fn=generate, inputs=inputs_generate, outputs=outputs_generate)
 process_interface = gr.Interface(fn=process, inputs=inputs_process, outputs=outputs_process)
 
-generate_interface.launch(share=True)
-process_interface.launch(share=True)
+interface = gr.Interface(
+    fn=generate,
+    inputs=gr.inputs.Dropdown(choices=list(models.keys()), label="Model"),
+    outputs=gr.outputs.Textbox(label="Generated Sentence")
+)
+
+interface.launch(share=True)