ArchitKohli commited on
Commit
f4c3f0d
·
1 Parent(s): f2f2c77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -178,7 +178,7 @@ def animize(img):
178
  prediction = (prediction[0]*0.5+0.5).numpy()
179
  out = tf.image.resize(prediction, [init_shape[0], init_shape[1]],method=tf.image.ResizeMethod.NEAREST_NEIGHBOR)
180
 
181
- return out
182
 
183
  interface = gr.Interface(fn=animize, inputs="image", outputs="image")
184
  interface.launch(share=True)
 
178
  prediction = (prediction[0]*0.5+0.5).numpy()
179
  out = tf.image.resize(prediction, [init_shape[0], init_shape[1]],method=tf.image.ResizeMethod.NEAREST_NEIGHBOR)
180
 
181
+ return out.numpy()
182
 
183
  interface = gr.Interface(fn=animize, inputs="image", outputs="image")
184
  interface.launch(share=True)