Update app.py

app.py

@@ -2,6 +2,9 @@ import os
 os.system('pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html')
 
 import gradio as gr
+import cv2
+os.system('pip install torch==1.9.0 torchvision==0.10.0')
+
 # check pytorch installation: 
 import torch, torchvision
 print(torch.__version__, torch.cuda.is_available())
@@ -12,7 +15,7 @@ import detectron2
 from detectron2.utils.logger import setup_logger
 # import some common libraries
 import numpy as np
-import os, json, cv2, random
+import os, json, random
 # import some common detectron2 utilities
 from detectron2 import model_zoo
 from detectron2.engine import DefaultPredictor