update gpu

Dockerfile

@@ -8,7 +8,7 @@ WORKDIR /app
 COPY . .
 
 # Install dependencies
-RUN pip install opencv-python fastapi uvicorn[standard] websockets
+RUN pip install "numpy<2" opencv-python fastapi uvicorn[standard] websockets
 
 ENV MODEL_PATH="model"
 ENV RESOLUTION="172"
@@ -17,4 +17,4 @@ ENV RESOLUTION="172"
 EXPOSE 8000
 
 # Chạy Server FastAPI
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]

inference.py

@@ -19,14 +19,14 @@ print("Model loaded!")
 def get_init_states():
     dummy = tf.zeros([1, 1, RESOLUTION, RESOLUTION, 3], dtype=tf.float32)
     return model.init_states(tf.shape(dummy))
-
+os.environ["OPENCV_FFMPEG_INTERRUPT_TIMEOUT"] = "60000"
 class VideoProcessor:
     def __init__(self):
         self.running = False
 
     def start_processing(self, rtsp_url, result_queue):
         self.running = True
-        cap = cv2.VideoCapture(rtsp_url)
+        cap = cv2.VideoCapture(rtsp_url, cv2.CAP_FFMPEG)
         if not cap.isOpened():
             result_queue.put({"error": "Cannot open RTSP URL"})
             return
@@ -100,4 +100,4 @@ class VideoProcessor:
             # Tùy chọn: Gửi Heartbeat mỗi 5s để biết model vẫn sống (nếu cần)
 
         cap.release()
-        result_queue.put({"status": "Stream stopped"})
+        result_queue.put({"status": "Stream stopped"})