https://huggingface.co/spaces/Saad123/minimal

app.py

@@ -1,13 +1,17 @@
 import gradio as gr
 import numpy as np
-from fastai.vision.all import *
+import tensorflow as tf
+import tensorflow_hub as hub
 
-classifier=load_learner("CatDogmodel.h5")
+loaded_model = tf.keras.models.load_model(
+       ('CatDogmodel.h5'),
+       custom_objects={'KerasLayer':hub.KerasLayer}
+)
 
 def model(image):
     im_scaled = image/255
     im_reshape = np.reshape(im_scaled,[1,224,224,3])
-    pred = classifier.predict(im_reshape)
+    pred = loaded_model.predict(im_reshape)
     if (pred == 0):
         return "The Image is of a Dog."
     if (pred == 1):