fix order of arguments
Browse files
app.py
CHANGED
|
@@ -68,7 +68,7 @@ with gr.Blocks() as demo:
|
|
| 68 |
plot = gr.Plot()
|
| 69 |
|
| 70 |
upload_btn.upload(fn=preprocess, inputs=[upload_btn, prediction_length, windows], outputs=plot)
|
| 71 |
-
train_btn.click(fn=train_and_forecast, inputs=[upload_btn, prediction_length,
|
| 72 |
|
| 73 |
if __name__ == "__main__":
|
| 74 |
demo.queue().launch()
|
|
|
|
| 68 |
plot = gr.Plot()
|
| 69 |
|
| 70 |
upload_btn.upload(fn=preprocess, inputs=[upload_btn, prediction_length, windows], outputs=plot)
|
| 71 |
+
train_btn.click(fn=train_and_forecast, inputs=[upload_btn, prediction_length, windows, epochs], outputs=plot)
|
| 72 |
|
| 73 |
if __name__ == "__main__":
|
| 74 |
demo.queue().launch()
|