Update app.py

app.py

@@ -10,6 +10,7 @@ from demo.predictors import VisualizationDemo
 from detectron2.checkpoint import DetectionCheckpointer
 from detectron2.config import LazyConfig, instantiate
 import numpy as np
+from PIL import Image
 
 if __name__ == "__main__":
     gr.close_all()
@@ -32,6 +33,7 @@ if __name__ == "__main__":
     def inference(img, confidence):
         img = np.array(img)
         _, results = vis_demo.run_on_image(img, confidence)
+        results = Image.fromarray(results.get_image())
         return results 
     
     demo = gr.Interface(