Update app.py

app.py

@@ -8,14 +8,14 @@ from ultralytics import YOLO
 model = YOLO('best.pt')
 
 # Function for image processing
-def show_preds_image(image_path):
+def show_det_image(image_path):
     image = cv2.imread(image_path)
     results = model.predict(source=image_path)
     annotated_image = results[0].plot()
     return cv2.cvtColor(annotated_image, cv2.COLOR_BGR2RGB)
 
 # Function for video processing
-def show_preds_video(video_path):
+def show_det_video(video_path):
     cap = cv2.VideoCapture(video_path)
     out_frames = []
     fps = int(cap.get(cv2.CAP_PROP_FPS))
@@ -42,21 +42,21 @@ def show_preds_video(video_path):
 inputs_image = gr.Image(type="filepath", label="Input Image")
 outputs_image = gr.Image(type="numpy", label="Output Image")
 interface_image = gr.Interface(
-    fn=show_preds_image,
+    fn=show_det_image,
     inputs=inputs_image,
     outputs=outputs_image,
     title="ITI107 Assignment - 4744695Y",
-    description="Taxi & License Plate Detection with Image"    
+    description="Taxi & License Plate Detection using YOLOv8"    
 )
 
 inputs_video = gr.Video(label="Input Video")
 outputs_video = gr.Video(label="Annotated Output")
 interface_video = gr.Interface(
-    fn=show_preds_video,
+    fn=show_det_video,
     inputs=inputs_video,
     outputs=outputs_video,
     title="ITI107 Assignment - 4744695Y",
-    description="Taxi & License Plate Detection with Video"
+    description="Taxi & License Plate Detection using YOLOv8"
 )
 
 gr.TabbedInterface(