Update app.py

app.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 import os
 import random
+import cv2
 
 import gradio as gr
 import numpy as np