Spaces:
Runtime error
Runtime error
add grad false part
Browse files
app.py
CHANGED
@@ -43,6 +43,8 @@ model.fc = nn.Sequential(
|
|
43 |
urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
|
44 |
model.load_state_dict(torch.load(MODEL_PATH))
|
45 |
model.eval()
|
|
|
|
|
46 |
|
47 |
inputs = gr.inputs.Image()
|
48 |
outputs = gr.outputs.Label(num_top_classes=1)
|
|
|
43 |
urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
|
44 |
model.load_state_dict(torch.load(MODEL_PATH))
|
45 |
model.eval()
|
46 |
+
for _, p in model.named_parameters():
|
47 |
+
p.requires_grad = False
|
48 |
|
49 |
inputs = gr.inputs.Image()
|
50 |
outputs = gr.outputs.Label(num_top_classes=1)
|