Update app.py

app.py

@@ -59,7 +59,7 @@ with gr.Blocks() as demo:
         number_input = gr.Textbox(label="Kol-vo textov")
         train_btn = gr.Button("Start Training", variant="primary")
         train_output = gr.Textbox(label="Result", interactive=False)
-        train_btn.click(train_model, inputs=[start_input,end_input] ,outputs=train_output)
+        train_btn.click(train_model, inputs=[number_input] ,outputs=train_output)
     
     with gr.Tab("Generate Text"):
         gr.Markdown("Generate Slovak text")