Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def model_parameter(model):
|
|
22 |
|
23 |
def predict(inp):
|
24 |
inp = transforms.ToTensor()(inp).unsqueeze(0)
|
25 |
-
model_parameter(
|
26 |
with torch.no_grad():
|
27 |
prediction = torch.nn.functional.softmax(model(inp)[0], dim=0)
|
28 |
confidences = {labels[i]: float(prediction[i]) for i in range(1000)}
|
|
|
22 |
|
23 |
def predict(inp):
|
24 |
inp = transforms.ToTensor()(inp).unsqueeze(0)
|
25 |
+
model_parameter(model)
|
26 |
with torch.no_grad():
|
27 |
prediction = torch.nn.functional.softmax(model(inp)[0], dim=0)
|
28 |
confidences = {labels[i]: float(prediction[i]) for i in range(1000)}
|