import tf.keras.backend

app.py

@@ -17,6 +17,7 @@ import gradio as gr
 from huggingface_hub import hf_hub_download
 from yolov5 import detect
 import numpy as np
+from tensorflow.keras import backend as K
 from utils import get_model, get_cfg, get_embeddings, get_comp_embeddings, get_test_embedding