Upload app.py

app.py

@@ -153,8 +153,8 @@ def analyze_image(image):
     
     # Run YOLO models
     run_yolo_models1(temp_image_path)
-    label_path = 'runs/predict-seg/yolov9_c_640_detect/labels/temp_image.txt'
-    label_path2 = 'runs/detect/yolov9_c_640_detect/labels/temp_image.txt'
+    label_path = '/runs/predict-seg/yolov9_c_640_detect/labels/temp_image.txt'
+    label_path2 = '/runs/detect/yolov9_c_640_detect/labels/temp_image.txt'
     
     segment = extract_class_0_coordinates(label_path)
     yolo_boxes = read_yolo_boxes(label_path2)
@@ -168,9 +168,9 @@ def analyze_image(image):
         results.append(f"{class_name} at ({x:.2f}, {y:.2f}) is {result} the segment.")
     
     # Clean up temporary files
-    #os.remove(temp_image_path)
-    #os.remove(label_path)
-    #os.remove(label_path2)
+    os.remove(temp_image_path)
+    os.remove(label_path)
+    os.remove(label_path2)
     
     return fig, "\n".join(results)