Update app.py

app.py

@@ -1,4 +1,3 @@
-from transformers import pipeline
 import gradio as gr
 import tensorflow as tf
 from PIL import Image
@@ -14,7 +13,7 @@ def predict_image(img):
     img_batch = np.expand_dims(img_array, axis=0)
     
     # Predict
-    pred = model.predict(img_batch)[0][0]
+    pred = model.predict(img_batch, verbose=0)[0][0]
     label = "dog" if pred > 0.5 else "cat"
     confidence = float(pred) if pred > 0.5 else float(1 - pred)
     return {label: confidence}
@@ -25,8 +24,8 @@ demo = gr.Interface(
     inputs=gr.Image(type="pil"),
     outputs=gr.Label(num_top_classes=2),
     examples=[],
-    title="Cat vs Dog Classifier",
-    description="Upload an image of a cat or dog. Trained on only 100 images!"
+    title="🐱 vs 🐶 Cat or Dog Classifier",
+    description="Trained on only 100 images! Upload a photo to see the prediction."
 )
 
 demo.launch()