Update app.py

app.py

@@ -3,12 +3,13 @@ from transformers import DetrImageProcessor, DetrForObjectDetection
 import torch
 import supervision as sv
 import json
+import requests
 
 image_processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
 model = DetrForObjectDetection.from_pretrained("Guy2/AirportSec-150epoch")
 id2label = {0: 'dangerous-items', 1: 'Gun', 2: 'Knife', 3: 'Pliers', 4: 'Scissors', 5: 'Wrench'}
-def anylize(img):
-    image = img
+def anylize(url):
+    image = Image.open(requests.get(url, stream=True).raw)
 
     with torch.no_grad():
 
@@ -28,4 +29,4 @@ def anylize(img):
     json_list = json.dumps(labels)
     return json_list
 
-gr.Interface(fn = anylize, inputs="image", outputs=gr.JSON()).launch()
+gr.Interface(fn = anylize, inputs="text", outputs=gr.JSON()).launch()