Create app.py

app.py

@@ -0,0 +1,21 @@
+from transformers import DetrFeatureExtractor, DetrForObjectDetection
+from PIL import Image
+import requests
+import gradio as gr
+
+def object_classify(img):
+  feature_extractor = DetrFeatureExtractor.from_pretrained('facebook/detr-resnet-50')
+  model = DetrForObjectDetection.from_pretrained('facebook/detr-resnet-50')
+  
+  inputs = feature_extractor(images=img, return_tensors="pt")
+  outputs = model(**inputs)
+  # model predicts bounding boxes and corresponding COCO classes
+  logits = outputs.logits
+  bboxes = outputs.pred_boxes
+  return logits, bboxes
+  
+ interface=gr.Interface(fn=imgtrans,
+                       inputs=gr.inputs.Image(shape=(224,224),label='Insert Image'),
+                       outputs=[gr.outputs.Textbox()],
+                       examples = [],description='??')
+interface.launch()