Update app.py

app.py

@@ -223,7 +223,15 @@ def run_diagnosis(backbone_name, input_image, preprocess_fn = None, Idx2labels=N
     # Generate predictions
     output_logits = model(input_tensor).cpu()
 
-    probabilities = torch.sigmoid(output_logits)[0].numpy().tolist()
+    # File "/app/app.py", line 226, in run_diagnosis
+    # probabilities = torch.sigmoid(output_logits)[0].numpy().tolist()
+    #                 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^
+    # RuntimeError: Can't call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.
+
+    #    ylogit = model(newimg).detach()
+    #     yprob = torch.sigmoid(ylogit)
+
+    probabilities = torch.sigmoid(output_logits)[0].detach().numpy().tolist()
 
     output_probs = dict()
     predicted_classes = []