Update app.py

app.py

@@ -37,7 +37,7 @@ def inference(input_image):
         model.to('cuda')
 
     with torch.no_grad():
-        output = model(input_batch)
+        output = model(input_tensor)
     # The output has unnormalized scores. To get probabilities, you can run a softmax on it.
     probabilities = torch.nn.functional.softmax(output[0])