Update app.py

app.py

@@ -6,11 +6,12 @@ from torchvision import models, transforms
 from PIL import Image
 
 # -- get torch and cuda version
-TORCH_VERSION = ".".join(torch.__version__.split(".")[:2])
-CUDA_VERSION = torch.__version__.split("+")[-1]
+#TORCH_VERSION = ".".join(torch.__version__.split(".")[:2])
+#CUDA_VERSION = torch.__version__.split("+")[-1]
 
 # -- install pre-build detectron2
-os.system(pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/{CUDA_VERSION}/{TORCH_VERSION}/index.html)
+os.system(pip install pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.10/index.html)
+os.system(pip install pyyaml==5.1)
 
 import detectron2
 from detectron2.utils.logger import setup_logger # ????