Update app.py

app.py

@@ -1,14 +1,16 @@
-import gradio as gr
 import matplotlib.pyplot as plt
 import numpy as np
+from six import BytesIO
 from PIL import Image
 import tensorflow as tf
 from object_detection.utils import label_map_util
 from object_detection.utils import visualization_utils as viz_utils
 from object_detection.utils import ops as utils_op
+import tarfile
+import wget 
 import gradio as gr
-import os
-import cv2
+from huggingface_hub import snapshot_download
+import os 
 
 def greet(name):
     return "Hello " + name + "!!"
@@ -52,12 +54,6 @@ detection_model = load_model()
 # predicted_img = predict(image_arr)
 # predicted_img.save('predicted.jpg')
 
-gr.Interface(fn=predict,
-             inputs=gr.Image(type="pil"),
-             outputs=gr.Image(type="pil")
-             ).launch(share=True)
-
-
 
 gr.Interface(fn=predict,
              inputs=[gr.Image(type="pil",label="Input Image"),gr.Textbox(placeholder="0.50",label="Set the confidence threshold (0.00-1.00)")],