ariG23498 commited on
Commit
85d2ea7
1 Parent(s): 884f790

chore: return a list of images from the infer function

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,11 +21,11 @@ def infer(test_image):
21
  inv2_kernel = tf.reduce_sum(inv2_kernel, axis=[-1, -2, -3])
22
  inv3_kernel = tf.reduce_sum(inv3_kernel, axis=[-1, -2, -3])
23
 
24
- return (
25
  tf.keras.utils.array_to_img(inv1_kernel[0, ..., None]),
26
  tf.keras.utils.array_to_img(inv2_kernel[0, ..., None]),
27
  tf.keras.utils.array_to_img(inv3_kernel[0, ..., None]),
28
- )
29
 
30
  # define the article
31
  article = """<center>
 
21
  inv2_kernel = tf.reduce_sum(inv2_kernel, axis=[-1, -2, -3])
22
  inv3_kernel = tf.reduce_sum(inv3_kernel, axis=[-1, -2, -3])
23
 
24
+ return [
25
  tf.keras.utils.array_to_img(inv1_kernel[0, ..., None]),
26
  tf.keras.utils.array_to_img(inv2_kernel[0, ..., None]),
27
  tf.keras.utils.array_to_img(inv3_kernel[0, ..., None]),
28
+ ]
29
 
30
  # define the article
31
  article = """<center>