Update app.py

app.py

@@ -7,7 +7,7 @@ def download_models(model_id):
     return f"./{model_id}"
 
 @spaces.GPU
-def yolov9_inference(img_path, model_id, image_size, conf_threshold, iou_threshold):
+def yolov8_inference(img_path, model_id, image_size, conf_threshold, iou_threshold):
     """
     Load a YOLOv9 model, configure it, perform inference on an image, and optionally adjust 
     the input size and apply test time augmentation.
@@ -78,7 +78,7 @@ def app():
                     step=0.1,
                     value=0.5,
                 )
-                yolov9_infer = gr.Button(value="Inference")
+                yolov8_infer = gr.Button(value="Inference")
 
             with gr.Column():
                 output_image = gr.Image(type="numpy",label="Output")