from ultralytics import YOLO
import os

# 1. Load your high-accuracy PC model
model_path = 'best.pt'
if not os.path.exists(model_path):
    print(f"Error: {model_path} not found.")
else:
    model = YOLO(model_path) 

    # Removing int8=True for debugging
    model.export(
        format='tflite', 
        nms=True, 
        imgsz=640
    )

    print("Mobile assets attempt finished.")