Update main.py

main.py

@@ -11,6 +11,7 @@ from utils import tts, stt, read_image_file, pil_to_base64, base64_to_pil, get_h
 import zipfile
 import soundfile as sf
 import openai
+import os
 
 # Config for camera picture
 model = YOLO('ultralyticsplus/yolov8s')