first commit

.gitignore

@@ -171,3 +171,9 @@ aaa.md
 img.jpg
 test_data
 testsrc.mp4
+
+*.jpg
+*.png
+*.pth
+.idea
+*.jpeg

Dockerfile

@@ -36,4 +36,5 @@ RUN pip install --no-cache-dir litserve==0.2.16
 RUN pip install -r requirements-GPU.txt
 RUN pip install opencv-python
 EXPOSE 8000
-CMD ["python", "/app/litserver_main.py"]
+#CMD ["python", "/app/litserver_main.py"]
+CMD ["tail", "-f","/dev/null"]

docker-compose.yml

@@ -0,0 +1,18 @@
+services:
+  lc_neo_refacer:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    working_dir: /app
+    volumes:
+      - .:/app
+    ports:
+      - "10071:8000"
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities:
+                - gpu

litserver_main.py

@@ -66,5 +66,5 @@ class InferencePipeline(ls.LitAPI):
 
 if __name__ == '__main__':
     api = InferencePipeline()
-    server = ls.LitServer(api, accelerator="gpu")
-    server.run(port=8080)
+    server = ls.LitServer(api, accelerator="auto")
+    server.run(port=8000)

requirements-GPU.txt

@@ -14,8 +14,8 @@ psutil==7.0.0
 ngrok==1.4.0
 pyfiglet==1.0.2
 # codeformer dependencies
-# torch==2.6.0
-# torchvision==0.21.0
+torch==2.6.0
+torchvision==0.21.0
 gdown==5.2.0
 lpips==0.1.4
 opencv-python