Update app.py

app.py

@@ -46,6 +46,6 @@ with gr.Blocks() as app:
     html_game = gr.HTML()
     
     update_game.click(game_fn,[out_url],html_game)
-    btn.click(dif_fn,prompt,out_im)
+    btn.click(dif_fn,prompt,[out_im,out_url])
     app.load(game_fn,None,html_game)
 app.launch()