Vvaann commited on
Commit
4f4cb25
·
verified ·
1 Parent(s): f4315f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -48,6 +48,7 @@ def resize_image_pil(image, new_width, new_height):
48
 
49
 
50
  def inference(input_img, show_gradcam, layer_name, num_classes, transparancy = 0.5):
 
51
  input_img = resize_image_pil(input_img,32,32)
52
  input_img = np.array(input_img)
53
  org_img = input_img
@@ -101,7 +102,7 @@ demo = gr.Interface(
101
  ],
102
  title = "CIFAR 10 trained on ResNet model in pytorch lightning with Gradcam",
103
  description = " A simple gradio inference to infer on resnet18 model",
104
- examples = [["cat.jpg", True, 10, "1", -1],
105
  ["dog.jpg", 4, False, "1", -1]]
106
  )
107
 
 
48
 
49
 
50
  def inference(input_img, show_gradcam, layer_name, num_classes, transparancy = 0.5):
51
+ print(show_gradcam, layer_name, num_classes, transparancy)
52
  input_img = resize_image_pil(input_img,32,32)
53
  input_img = np.array(input_img)
54
  org_img = input_img
 
102
  ],
103
  title = "CIFAR 10 trained on ResNet model in pytorch lightning with Gradcam",
104
  description = " A simple gradio inference to infer on resnet18 model",
105
+ examples = [["cat.jpg", 10, True, "1", -1],
106
  ["dog.jpg", 4, False, "1", -1]]
107
  )
108