Shivdutta commited on
Commit
09b5b07
·
verified ·
1 Parent(s): 78fa729

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -83,8 +83,8 @@ def inference(input_img, num_gradcam_images=1, target_layer_number=-1, transpare
83
  else:
84
  return top_n_confidences, visualization, None
85
 
86
- title = "CIFAR10 trained on ResNet18 Model with GradCAM"
87
- description = "A simple Gradio interface to infer on ResNet model, and get GradCAM results"
88
  examples = [["cat.jpg", 1, -1, 0.8, True, 3, 3],
89
  ["dog.jpg", 1, -1, 0.8, True, 3, 3],
90
  ["plane.jpg", 1, -1, 0.8, True, 3, 3],
 
83
  else:
84
  return top_n_confidences, visualization, None
85
 
86
+ title = "CIFAR10 trained on ResNet18 Model using Pytorch Lightning with GradCAM"
87
+ description = "A simple Gradio interface to infer on ResNet18 model using Pytorch Lightning, and get GradCAM results"
88
  examples = [["cat.jpg", 1, -1, 0.8, True, 3, 3],
89
  ["dog.jpg", 1, -1, 0.8, True, 3, 3],
90
  ["plane.jpg", 1, -1, 0.8, True, 3, 3],