build: cleanup

app.py

@@ -21,7 +21,6 @@ import gradio as gr
 import huggingface_hub
 import numpy as np
 import PIL.Image
-import torch
 
 import anime_face_detector
 
@@ -81,9 +80,7 @@ def main():
     parser.add_argument(
         "--detector", type=str, default="yolov3", choices=["yolov3", "faster-rcnn"]
     )
-    parser.add_argument(
-        "--device", type=str, default="cpu", choices=["cuda:0", "cpu"]
-    )
+    parser.add_argument("--device", type=str, default="cpu", choices=["cuda:0", "cpu"])
     parser.add_argument("--face-score-threshold", type=float, default=0.5)
     parser.add_argument("--landmark-score-threshold", type=float, default=0.3)
     parser.add_argument("--score-slider-step", type=float, default=0.05)

requirements.txt

@@ -1,4 +1,4 @@
-# python=3.10
+# python=3.8
 openmim==0.3.7
 mmcv-full==1.6.2
 mmdet==2.28.2
@@ -9,6 +9,9 @@ scipy==1.10.1
 
 opencv-python-headless==4.7.0.72
 
+# PyTorch with CUDA support (use pip with `--index-url https://download.pytorch.org/whl/cu118`)
+# torch==2.0.1+cu118
+
 torch==2.0.1
 torchvision==0.15.2