Spaces:

keras-io
/

patch-conv-net

Runtime error

ariG23498 HF Staff commited on Jan 24, 2022

Commit

0469f00

1 Parent(s): 532e265

fix: change the call method signature

Files changed (1) hide show

utilities/visualization.py CHANGED Viewed

@@ -8,7 +8,7 @@ class PlotAttention(object):
 	def __init__(self, model):
 		self.model = model
-	def  __call__(self, image):
 		# resize the image to a 224, 224 dim
 		image = tf.image.convert_image_dtype(image, tf.float32)
 		image = tf.image.resize(image, (224, 224))

 	def __init__(self, model):
 		self.model = model
+	def __call__(self, image):
 		# resize the image to a 224, 224 dim
 		image = tf.image.convert_image_dtype(image, tf.float32)
 		image = tf.image.resize(image, (224, 224))