phillipinseoul commited on
Commit
8f04382
1 Parent(s): 12a81fb

fix output

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -4,6 +4,7 @@ An interactive demo for text-guided panorama generation.
4
  """
5
  import os
6
  from os.path import join
 
7
  import torch
8
  import gradio as gr
9
 
@@ -78,9 +79,9 @@ if __name__=="__main__":
78
  ],
79
  inputs=[prompt, width, sync_weight, sync_thres, seed],
80
  outputs=[
81
- join(os.path.dirname(__file__), "assets", "result_castle_seed_1.png"),
82
- join(os.path.dirname(__file__), "assets", "result_natural_seed_2.png"),
83
- join(os.path.dirname(__file__), "assets", "result_northern_seed_6.png"),
84
  ]
85
  )
86
 
 
4
  """
5
  import os
6
  from os.path import join
7
+ from PIL import Image
8
  import torch
9
  import gradio as gr
10
 
 
79
  ],
80
  inputs=[prompt, width, sync_weight, sync_thres, seed],
81
  outputs=[
82
+ [gr.Image(Image.open(join(os.path.dirname(__file__), "assets", "result_castle_seed_1.png")))],
83
+ [gr.Image(Image.open(join(os.path.dirname(__file__), "assets", "result_natural_seed_2.png")))],
84
+ [gr.Image(Image.open(join(os.path.dirname(__file__), "assets", "result_northern_seed_6.png")))],
85
  ]
86
  )
87