dusaurabh commited on
Commit
4b7c2a1
1 Parent(s): 28ed5fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -39,7 +39,9 @@ def resize_image_pil(image, new_width, new_height):
39
 
40
  return resized
41
 
42
- def inference(input_img, transparency = 0.5, target_layer_number = -1, grad_cam_option = "Yes", no_of_classes_view = 3):
 
 
43
  input_img = resize_image_pil(input_img, 32, 32)
44
 
45
  input_img = np.array(input_img)
 
39
 
40
  return resized
41
 
42
+ no_of_classes_view = 3
43
+
44
+ def inference(input_img, transparency = 0.5, target_layer_number = -1, grad_cam_option = "Yes", no_of_classes_view = no_of_classes_view):
45
  input_img = resize_image_pil(input_img, 32, 32)
46
 
47
  input_img = np.array(input_img)