Hugging Face's logo

Spaces:
debisoft
/
wcgan-a
Sleeping

App Files Community
debisoft commited on
Commit
f40ad78
·
1 Parent(s): 399a541

Handle extra dimension

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def greet(input):
30
  pixel_values = transform(image).unsqueeze(0)
31
  output = generator(pixel_values)
32
 
33
- return transform2(output)
34
 
35
  iface = gr.Interface(fn=greet, inputs=[gr.Textbox(label="coord_zxy", value="/18/73237/95677")], outputs=[gr.Image(type="pil", width=256, label="Output Image")])
36
 
 
30
  pixel_values = transform(image).unsqueeze(0)
31
  output = generator(pixel_values)
32
 
33
+ return transform2(output[0])
34
 
35
  iface = gr.Interface(fn=greet, inputs=[gr.Textbox(label="coord_zxy", value="/18/73237/95677")], outputs=[gr.Image(type="pil", width=256, label="Output Image")])
36