Update predict.py

predict.py

@@ -3,6 +3,10 @@ import random
 import numpy as np
 from PIL import Image
 from collections import defaultdict
+import os
+# install detectron this way to avoid torch not available error if mentioned directly as dependency in requirements.txt
+os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
+
 from detectron2.data import MetadataCatalog
 from detectron2.utils.visualizer import ColorMode, Visualizer
 from color_palette import ade_palette