Update app.py

app.py

@@ -6,8 +6,10 @@ import tensorflow as tf
 import tensorflow_hub as hub
 from tensorflow.keras.models import load_model
 import os
+import cv2
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+
 # Specify the custom objects for loading the model
 custom_objects = {'KerasLayer': hub.KerasLayer}
 
@@ -44,5 +46,4 @@ input_image = gr.Image(shape=(224, 224), label="Input Image")
 output_label = gr.Label(label="Predicted Bird Species")
 
 # Launch Gradio
-gr.Interface(fn=predict_image, inputs=input_image, outputs=output_label, capture_session=True).launch()
-
+gr.Interface(fn=predict_image, inputs=input_image, outputs=output_label, capture_session=True).launch()