Changed medseg.py file

app/mediseg.py

@@ -297,9 +297,8 @@ def process_pneumonia(image: Image.Image, model_path=os.path.join("models", "pne
                              std=[0.229,0.224,0.225])
     ])
     input_tensor = transform_img(image).unsqueeze(0).to(device)
-    with torch.no_grad():
-        cam, output = grad_cam.generate(input_tensor)
-        predicted_class = output.argmax(dim=1).item()
+    cam, output = grad_cam.generate(input_tensor)
+    predicted_class = output.argmax(dim=1).item()
     label_text = "Pneumonia" if predicted_class == 1 else "Normal"
     
     def get_bounding_box(heatmap, thresh=0.5, min_area=100):