Update app.py

app.py

@@ -44,7 +44,7 @@ def preprocess_nii(nii_file):
 def predict(input_image):
     x = preprocess_nii(input_image)
     with torch.no_grad():
-        output = model(x)
+        logits = model(x)
         probs = F.softmax(logits, dim=1)  # convert to probabilities
         
     results = {cls: float(prob) for cls, prob in zip(CLASSES, probs)}