fix cuda

app.py

@@ -63,7 +63,7 @@ def resize_fix_aspect_ratio(img, field, target_width=None, target_height=None):
 def inference(img_rgb, model_type):
     if model_type is None:
         return None, ""
-    pf_model = PerspectiveFields(model_type).eval().cuda()
+    pf_model = PerspectiveFields(model_type).eval().to(device)
     pred = pf_model.inference(img_bgr=img_rgb[...,::-1])
     img_h = img_rgb.shape[0]
     field = {

setup.py

@@ -1,40 +0,0 @@
-from setuptools import setup, find_packages
-
-setup(
-    name='debug',
-    version='0.1.0',
-    packages=find_packages(),
-    install_requires=[
-        'torch==1.11.0+cu113',
-        'torchvision==0.12.0+cu113',
-        'mmcv==2.0.0rc4',
-        'albumentations==1.3.0',
-        'fvcore==0.1.5.post20221221',
-        'gradio==3.29.0',
-        'h5py==3.8.0',
-        'imageio==2.27.0',
-        'matplotlib==3.7.1',
-        'numpy==1.24.2',
-        'opencv_contrib_python==4.7.0.72',
-        'opencv_python==4.7.0.72',
-        'opencv_python_headless==4.7.0.72',
-        'Pillow==9.5.0',
-        'pycocotools==2.0.6',
-        'pyequilib==0.3.0',
-        'scikit-learn==1.2.2',
-        'scipy==1.9.1',
-        'setuptools==59.5.0',
-        'timm==0.6.13',
-        'tqdm==4.65.0',
-    ],
-    dependency_links=[
-        'git+https://github.com/facebookresearch/detectron2.git@4e447553eb32b6e3784df0b8fca286935107b2fd',
-        'git+https://github.com/jinlinyi/PerspectiveFields.git@dev#egg=perspective2d'
-    ],
-    extras_require={
-        'cu113': [
-            '--extra-index-url https://download.pytorch.org/whl/cu113',
-            '--find-links https://download.openmmlab.com/mmcv/dist/cu113/torch1.11/index.html'
-        ]
-    }
-)