Update app.py

app.py

@@ -2,7 +2,7 @@ from keras.models import load_model
 import cv2
 from tensorflow.keras.preprocessing.image import ImageDataGenerator
 import gradio as gr
-
+import numpy as np
 
 heart_model=load_model('Chicken_Heart_model.h5',compile=True)
 class_name={0:'Dilation(eccentric)',1:'Hepatoma',2:'Hypertrophy(concentric)',3:'Hypertrophy(physiological)',4:'Infraction Damage',5:'Normal'}