Create docker-compose.yml

docker-compose.yml

@@ -0,0 +1,22 @@
+version: '3.8'
+
+services:
+  huggingface-android:
+    build: .
+    image: huggingface-android:latest
+    container_name: huggingface-android
+    environment:
+      - HUGGINGFACE_TOKEN=${HUGGINGFACE_TOKEN}
+    volumes:
+      - ./models:/workspace/models
+      - ./android-project:/workspace/android-project
+      - ~/.gradle:/root/.gradle
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
+    tty: true
+    stdin_open: true