Update predict.py

predict.py

@@ -4,6 +4,7 @@ import numpy as np
 from PIL import Image
 from collections import defaultdict
 import os
+import spaces
 # Mentioning detectron2 as a dependency directly in requirements.txt tries to install detectron2 before torch and results in an error even if torch is listed as a dependency before detectron2.
 # Hence, installing detectron2 this way when using Gradio HF spaces.
 os.system('pip install git+https://github.com/facebookresearch/detectron2.git')