Update app.py

app.py

@@ -14,6 +14,12 @@ from detectron2.config import get_cfg
 from detectron2.utils.visualizer import Visualizer
 from detectron2.data import MetadataCatalog
 
+from matplotlib.pyplot import axis
+from torch import nn
+import requests
+
+import torch
+
 # Predefined Detectron2 models
 models = [
     {