Update app.py

app.py

@@ -5,7 +5,7 @@ import json
 import zipfile
 from tools.demo_api import build_predictor, run_detection
 import shutil
-import square_crop
+from square_crop import run_square_crop
 
 PREDICTOR = build_predictor(
     exp_file = "exps/yolox_s.py",
@@ -43,7 +43,7 @@ def process_yolox_api(files):
     try:
         
         coco_json_path = run_detection(PREDICTOR, input_dir)
-        cropped_paths = square_crop(input_dir, coco_json_path, cropped_dir)
+        cropped_paths = run_square_crop(input_dir, coco_json_path, cropped_dir)
         
         # Create zip file
         zip_path = os.path.join(temp_dir, "cropped_results.zip")