Update app.py

app.py

@@ -9,7 +9,7 @@ og_model = 'facebook/detr-resnet-50'
 image_processor =  DetrImageProcessor.from_pretrained(og_model)
 model = AutoModel.from_pretrained("taroii/notfinetuned-detr-50")
 
-def query(image):
+def predict(image):
     with torch.no_grad():
         # load image and predict
         inputs = image_processor(images=image, return_tensors='pt')
@@ -28,8 +28,13 @@ def query(image):
     labels = [f"{id2label[class_id]} {confidence:.2f}" for _, confidence, class_id, _ in detections]
     frame = box_annotator.annotate(scene=image.copy(), detections=detections, labels=labels)
     
-    plot_image(frame, (16, 16))
+    return plot_image(frame, (16, 16))
     
-    return labels, frame
+    #return labels
 
-gr.Interface.load("models/taroii/notfinetuned-detr-50").launch()
+gr.Interface(
+    predict,
+    inputs=gr.inputs.Image(label="Upload hot dog candidate", type="filepath"),
+    outputs=gr.JSON(),
+    title="Non-Fine-Tuned Model"
+).launch()