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

CatDogmodel2.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3fdef209df3115267f8318110fc0c814c927d2f7fd3fb7a7a7f893e42690d064
+size 87673584

app.py

@@ -1,6 +1,24 @@
 import gradio as gr
-def greet(name):
-    return "Hello " + name + "!!"
+import numpy as np
+import tensorflow as tf
+import tensorflow_hub as hub
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+loaded_model = tf.keras.models.load_model(
+       ('CatDogmodel2.h5'),
+       custom_objects={'KerasLayer':hub.KerasLayer}
+)
+
+def model(image):
+    im_scaled = image/255
+    im_reshape = np.reshape(im_scaled,[1,160,160,3])
+    pred = loaded_model.predict(im_reshape)
+    pred_label = np.argmax(pred)
+    if (pred_label == 0):
+        return "The Image is of a Dog."
+    if (pred_label == 1):
+        return "The Image is of a Cat."
+
+image = gr.inputs.Image(shape=(160,160))
+
+iface = gr.Interface(fn=model, inputs=image, outputs='text')
+iface.launch(debug=True)

requirements.txt

@@ -0,0 +1,4 @@
+gradio==3.1.7
+numpy==1.21.5
+tensorflow==2.9.1
+tensorflow_hub==0.12.0