modify app.py

app.py

@@ -34,7 +34,7 @@ def process_video(video_file):
             # Save the frame to the temporary directory
             frame_path = os.path.join(temp_dir, f"frame_{frame_count}.jpg")
             cv2.imwrite(frame_path, frame)
-            results = model.predict(source=frame_path, conf=0.002, save=False, show=False, line_width=2, show_labels=True, show_boxes=True, retina_masks=True)
+            results = model.predict(source=frame_path, conf=0.2, save=False, show=False, line_width=2, show_labels=True, show_boxes=True, retina_masks=True)
             for result in results:
                 boxes = result.boxes
                 # result.save(filename=frame_path)
@@ -62,9 +62,9 @@ def process_video(video_file):
 iface = gr.Interface(
     fn=process_video,  # Function to process the video and return extracted frames
     inputs=gr.Video(label="Upload your video"),  # Video upload input
-    outputs=gr.Gallery(label="Extracted Frames"),  # Display extracted frames in a gallery
-    title="Video Frame Extraction",  # Interface title
-    description="Upload a video, extract 5 frames per second, resize if necessary, and display extracted frames."
+    outputs=gr.Gallery(label="Output Images Results"),  # Display extracted frames in a gallery
+    title="Road Damage Detection"  # Interface title
+    # description="Upload a video, extract 5 frames per second, resize if necessary, and display extracted frames."
 )
 
 # Launch the Gradio app