amitkayal commited on
Commit
144a4d9
·
1 Parent(s): babbaf9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -77,6 +77,7 @@ def inference_visualization(input_img, transparency = 0.5, target_layer_number =
77
  title = "CIFAR10 trained on ResNet18 Model with GradCAM"
78
  description = "Gradio interface to infer on ResNet18 model, and get GradCAM results"
79
  examples = [["cat.jpg", 0.5, -1], ["dog.jpg", 0.5, -1]]
 
80
  demo = gr.Interface(
81
  # inference,
82
  # inputs = [gr.Image(shape=(32, 32), label="Input Image"), gr.Slider(0, 1, value = 0.5, label="Opacity of GradCAM"), gr.Slider(-2, -1, value = -2, step=1, label="Which Layer?")],
@@ -84,7 +85,7 @@ demo = gr.Interface(
84
  # title = title,
85
  # description = description,
86
  # examples = examples,
87
- fn=inference, # We'll add the function later after defining all functions
88
  inputs=[
89
  gr.Radio(["Yes", "No"], label="View GradCAM images?"),
90
  gr.Number(label="Number of GradCAM images to view", default=5, max=10),
 
77
  title = "CIFAR10 trained on ResNet18 Model with GradCAM"
78
  description = "Gradio interface to infer on ResNet18 model, and get GradCAM results"
79
  examples = [["cat.jpg", 0.5, -1], ["dog.jpg", 0.5, -1]]
80
+
81
  demo = gr.Interface(
82
  # inference,
83
  # inputs = [gr.Image(shape=(32, 32), label="Input Image"), gr.Slider(0, 1, value = 0.5, label="Opacity of GradCAM"), gr.Slider(-2, -1, value = -2, step=1, label="Which Layer?")],
 
85
  # title = title,
86
  # description = description,
87
  # examples = examples,
88
+ fn=None, # We'll add the function later after defining all functions, # We'll add the function later after defining all functions
89
  inputs=[
90
  gr.Radio(["Yes", "No"], label="View GradCAM images?"),
91
  gr.Number(label="Number of GradCAM images to view", default=5, max=10),