Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ import itertools
|
|
| 10 |
import matplotlib.pyplot as plt
|
| 11 |
from utils import LitCIFAR10
|
| 12 |
|
| 13 |
-
model = LitCIFAR10.load_from_checkpoint("model.ckpt")
|
| 14 |
model.eval()
|
| 15 |
|
| 16 |
classes = ('plane', 'car', 'bird', 'cat',
|
|
@@ -120,7 +120,7 @@ def inference(wants_gradcam, n_gradcam, target_layer_number, transparency, wants
|
|
| 120 |
|
| 121 |
title = "CIFAR10 trained on Custom ResNet Model with GradCAM"
|
| 122 |
description = "A simple Gradio interface to infer on ResNet model, and get GradCAM results"
|
| 123 |
-
examples = [[None, None, None, None, None, None, 'test_'+str(i)+'.jpg', None] for i in range(10)]
|
| 124 |
|
| 125 |
demo = gr.Interface(inference,
|
| 126 |
inputs = [gr.Checkbox(False, label='Do you want to see GradCAM outputs?'),
|
|
|
|
| 10 |
import matplotlib.pyplot as plt
|
| 11 |
from utils import LitCIFAR10
|
| 12 |
|
| 13 |
+
model = LitCIFAR10.load_from_checkpoint("model/model.ckpt")
|
| 14 |
model.eval()
|
| 15 |
|
| 16 |
classes = ('plane', 'car', 'bird', 'cat',
|
|
|
|
| 120 |
|
| 121 |
title = "CIFAR10 trained on Custom ResNet Model with GradCAM"
|
| 122 |
description = "A simple Gradio interface to infer on ResNet model, and get GradCAM results"
|
| 123 |
+
examples = [[None, None, None, None, None, None, 'Images/test_'+str(i)+'.jpg', None] for i in range(10)]
|
| 124 |
|
| 125 |
demo = gr.Interface(inference,
|
| 126 |
inputs = [gr.Checkbox(False, label='Do you want to see GradCAM outputs?'),
|