JSun22 commited on
Commit
a780e38
1 Parent(s): 1ada4b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(mdoel)
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)}