fix variable name

app.py

@@ -29,7 +29,7 @@ def query_image(img):
  with torch.no_grad():
  #preds = model(inputs.unsqueeze(0).to(device))["logits"]
  preds = model(**inputs)["logits"]
- preds_upscale = upscale_logits(preds, image.shape[2])
+ preds_upscale = upscale_logits(preds, img.shape[2])
  predict_label = torch.argmax(preds_upscale, dim=1).to(device)
  return predict_label[0,:,:].detach().cpu().numpy()