StevenChen16 commited on
Commit
b3fa7a7
·
verified ·
1 Parent(s): eeb2f25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -38,7 +38,6 @@ def process_images(content_img, style_img, epochs, steps_per_epoch, learning_rat
38
  output_img = main(content_img, style_img, epochs, steps_per_epoch, learning_rate, content_loss_factor, style_loss_factor, img_size, img_width, img_height)
39
  return output_img
40
 
41
-
42
  @spaces.GPU
43
  def create_app():
44
  with gr.Blocks() as demo:
@@ -84,7 +83,8 @@ def create_app():
84
  outputs=[output_img]
85
  )
86
 
87
- demo.launch()
88
 
89
  if __name__ == "__main__":
90
- create_app()
 
 
38
  output_img = main(content_img, style_img, epochs, steps_per_epoch, learning_rate, content_loss_factor, style_loss_factor, img_size, img_width, img_height)
39
  return output_img
40
 
 
41
  @spaces.GPU
42
  def create_app():
43
  with gr.Blocks() as demo:
 
83
  outputs=[output_img]
84
  )
85
 
86
+ return demo
87
 
88
  if __name__ == "__main__":
89
+ app = create_app()
90
+ app.launch()