Update main.py

main.py

@@ -11,6 +11,9 @@ from utils import tts, read_image_file, pil_to_base64, get_hist
 from typing import Optional
 from huggingface_hub import hf_hub_download
 
+from io import BytesIO
+import zipfile
+
 model_path = hf_hub_download(repo_id="ultralyticsplus/yolov8s", filename='yolov8s.pt')
 model = YOLO(model_path)
 CLASS = model.model.names