Update main.py

main.py

@@ -17,8 +17,8 @@ import uvicorn
 from skimage import color
 import hashlib
 
-# Speed up YOLO for Hugging Face's 2-core CPUs
-torch.set_num_threads(2)
+# OOM PREVENTION
+torch.set_num_threads(1)
 
 # --- CONFIGURATION ---
 MODEL_PATH = "best.pt"
@@ -596,12 +596,7 @@ class WatermelonProcessor:
                 cv2.circle(output, pt_bot, 8, (255, 255, 255), -1)
 
             cv2.polylines(output, [fit_pts.astype(np.int32)], True, (0, 255, 0), 3)
-            # --- SPEED FIX: Shrink the preview image to eliminate Network/Base64 Lag ---
-            prev_h, prev_w = output.shape[:2]
-            if max(prev_h, prev_w) > 640:
-                p_scale = 640.0 / max(prev_h, prev_w)
-                output = cv2.resize(output, (int(prev_w * p_scale), int(prev_h * p_scale)), interpolation=cv2.INTER_AREA)
-
+            
             _, buffer = cv2.imencode(".jpg", output, [cv2.IMWRITE_JPEG_QUALITY, 80])
             img_base64 = base64.b64encode(buffer).decode("utf-8")
         mark("render")