Update app.py

app.py

@@ -35,7 +35,6 @@ labels = {0: 'bacterial_leaf_blight',
 def inference_fn(model, image=None):  
     model.eval()
     image = image.to(device)  
-    print(image.shape)
     with torch.no_grad():
         output = model(image.unsqueeze(0))
     out = output.sigmoid().detach().cpu().numpy().flatten()   
@@ -43,7 +42,7 @@ def inference_fn(model, image=None):
     
     
 def predict(image = None) :  
-    image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+    #image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     mean = (0.485, 0.456, 0.406) 
     std = (0.229, 0.224, 0.225)