Update app.py

app.py

@@ -11,7 +11,7 @@ from datetime import datetime
 # Setup logging
 logging.basicConfig(level=logging.INFO)
 
-# Salesforce Credentials
+# Salesforce Credentials (ensure these are set up in your environment variables for security)
 SALESFORCE_USERNAME = "drone@sathkrutha.com"
 SALESFORCE_PASSWORD = "Komal1303@"
 SALESFORCE_SECURITY_TOKEN = "53AWRskW9EjWUsSL5LU6nFTy3"
@@ -33,7 +33,7 @@ except Exception as e:
     logging.error(f"Failed to connect to Salesforce: {str(e)}")
     raise Exception(f"Failed to connect to Salesforce: {str(e)}")
 
-# Load Model
+# Load Model (updated to use correct weights for the current PyTorch version)
 model = models.detection.fasterrcnn_resnet50_fpn(weights="FasterRCNN_ResNet50_FPN_Weights.COCO_V1")
 model.eval()
 
@@ -114,7 +114,7 @@ def detect_defects(image):
 
         for i in range(len(predictions[0]['boxes'])):
             score = predictions[0]['scores'][i].item()
-            if score < 0.5:  # Lowered threshold for testing
+            if score < 0.3:  # Lowered threshold to detect more objects
                 continue
 
             box = predictions[0]['boxes'][i].tolist()