Update app.py

app.py

@@ -25,6 +25,7 @@ vegetables = [
 model = YOLO('fresh_model.pt')
 # Load the YOLO model
 yolo_model = YOLO('another_model.pt')  # Adjust the path as needed
+CONFIDENCE_THRESHOLD=0.6
 
 @app.route('/')
 def index():
@@ -54,7 +55,7 @@ def predict():
                 # Use model's class names dynamically
                 vegetable_name = model.names[class_index]  # Get the name directly from the model
                 detected_vegetables.append(vegetable_name)
-        detected_classes=str(detected_vegetables[0])
+        detected_classes=detected_vegetables
 
         return jsonify({'predictions': detected_classes})
     except Exception as e:
@@ -93,7 +94,7 @@ def detect():
 
 
 
-model = YOLO('model_2.pt')  # Replace with your model's path
+modeltext = YOLO('model_2.pt')  # Replace with your model's path
 
 @app.route('/ocr', methods=['POST'])
 def ocr():
@@ -105,7 +106,7 @@ def ocr():
         image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
 
         # Perform text detection using YOLO
-        results = model(image)
+        results = modeltext(image)
         detections = results[0].boxes
         high_conf_detections = [box for box in detections if box.conf > 0.55]