yuhe6 commited on
Commit
c09d9df
·
1 Parent(s): 165b6ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def inference(input_image):
37
  model.to('cuda')
38
 
39
  with torch.no_grad():
40
- output = model(input_batch)
41
  # The output has unnormalized scores. To get probabilities, you can run a softmax on it.
42
  probabilities = torch.nn.functional.softmax(output[0])
43
 
 
37
  model.to('cuda')
38
 
39
  with torch.no_grad():
40
+ output = model(input_tensor)
41
  # The output has unnormalized scores. To get probabilities, you can run a softmax on it.
42
  probabilities = torch.nn.functional.softmax(output[0])
43