Update app.py

app.py

@@ -47,7 +47,7 @@ def generate_text(tokenizer, model, device, prompt, max_length=100,
     )
 
     # Convert the generated text back to a string
-    generated_text = [tokenizer.decode(ids, skip_special_tokens=True) for ids in output]
+    generated_text = tokenizer.decode(ids, skip_special_tokens=True) for ids in output
 
     return generated_text
 
@@ -77,13 +77,13 @@ tokenizer, model, device = load_model_and_tokenizer(model_path)
 iface = gr.Interface(
     fn=gradio_generate_text,
     inputs=[
-        gr.Textbox(lines=5, label="Input Prompt"),
+        gr.Textbox(lines=3, label="Input Prompt"),
         gr.Slider(minimum=10, maximum=768, value=32, label="Max Length"),
         gr.Slider(minimum=1, maximum=5, value=1, label="Number of Sequences"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.95, label="Top-p Sampling"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.7, label="Temperature")
     ],
-    outputs=gr.Textbox(label="Generated Text"),
+    outputs=gr.Textbox(lines=5, label="Generated Text"),
     title="Vietnamese Medical T5 Fine-Tuned Model",
     description="Generate text using a fine-tuned Vietnamese medical T5 model."
 )