vishnun commited on
Commit
2577b16
1 Parent(s): a09ee7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ import matplotlib.pyplot as plt
7
 
8
  i1 = gr.inputs.Image(shape=(256, 256))
9
  #i2 = gr.inputs.Slider(minimum=2, maximum=4, step=0.1, default=None, label="Scale for intensity - the more value the less the intensity in the pixels")
10
- o1 = gr.outputs.Image()
11
- o2 = gr.outputs.Image()
12
  gen_model = load_model('256_model_250ep.h5')
13
 
14
  def colorify(pixels):
 
7
 
8
  i1 = gr.inputs.Image(shape=(256, 256))
9
  #i2 = gr.inputs.Slider(minimum=2, maximum=4, step=0.1, default=None, label="Scale for intensity - the more value the less the intensity in the pixels")
10
+ o1 = gr.outputs.Image(type='numpy')
11
+ o2 = gr.outputs.Image(type='numpy')
12
  gen_model = load_model('256_model_250ep.h5')
13
 
14
  def colorify(pixels):