Update app.py

app.py

@@ -101,7 +101,10 @@ def predict_video(input_video: str):
             processed_frame = cv2.resize(processed_frame, (width, height))
             
             # Add text with prediction and confidence
-            text = f"{prediction}: {confidence:.2f}"
+            if prediction is not None and confidence is not None:
+                text = f"{prediction}: {confidence:.2f}"
+            else:
+                text = "No prediction available"
             cv2.putText(processed_frame, text, (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
             
             out.write(processed_frame)