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?'),
|