Upload app.py

app.py

@@ -1,11 +1,11 @@
 try:
     import detectron2
 except:
-    import os 
+    import os
     os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
 
 import cv2
-
+import torch
 from matplotlib.pyplot import axis
 import gradio as gr
 import requests
@@ -66,7 +66,7 @@ gr.Interface(
     [gr.inputs.Textbox(label="Image URL", placeholder="https://api.digitale-sammlungen.de/iiif/image/v2/bsb10483966_00008/full/500,/0/default.jpg"),
      gr.inputs.Image(type="numpy", label="Input Image"),
      gr.Slider(minimum=0.0, maximum=1.0, value=0.5, label="Minimum score"),
-    ], 
+    ],
     gr.outputs.Image(type="pil", label="Output"),
     title=title,
     description=description,