diff --git a/docker-compose.yml b/docker-compose.yml
index 3db56d6..1a8090d 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -12,4 +12,11 @@ services:
       - "5001"
     restart: unless-stopped
     volumes:
-      - ./backend/uploads:/app/backend/uploads
\ No newline at end of file
+      - ./backend/uploads:/app/backend/uploads
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
\ No newline at end of file