fix req 9

main.py

@@ -1,8 +1,14 @@
 import os
-import sys
 import tempfile
+
+# Set up Hugging Face cache directory
+os.environ['TRANSFORMERS_CACHE'] = os.path.join(tempfile.gettempdir(), 'huggingface_cache')
+os.makedirs(os.environ['TRANSFORMERS_CACHE'], exist_ok=True)
+
+# Now import the rest of the dependencies
+import sys
 import logging
-import shutil
+from pathlib import Path
 from flask import Flask, request, jsonify, render_template, send_file, Response
 from werkzeug.utils import secure_filename
 import cv2

requirements.txt

@@ -3,6 +3,7 @@ flask==2.0.1
 torch==2.0.0
 torchvision==0.15.1
 transformers==4.30.2
+huggingface_hub==0.13.4
 sentence-transformers==2.2.2
 opencv-python==4.5.3.56
 flask-cors==3.0.10