Update app.py

app.py

@@ -6,6 +6,9 @@ from supervision.detection.annotate import BoxAnnotator
 from supervision.utils.notebook import plot_image
 import cv2
 
+CONFIDENCE_TRESHOLD = 0.5
+IOU_TRESHOLD = 0.8
+
 og_model = 'facebook/detr-resnet-50'
 image_processor =  DetrImageProcessor.from_pretrained(og_model)
 model = AutoModel.from_pretrained("taroii/notfinetuned-detr-50")
@@ -36,7 +39,7 @@ def predict(image_path):
 
 gr.Interface(
     predict,
-    inputs=gr.inputs.Image(label="Upload hot dog candidate", type="filepath"),
+    inputs=gr.inputs.Image(label="Upload X-Ray Image", type="filepath"),
     outputs=gr.Image(),
     title="Non-Fine-Tuned Model"
 ).launch()