Update app.py

app.py

@@ -2,15 +2,14 @@ import gradio as gr
 import tensorflow as tf
 import numpy as np
 from PIL import Image
-import spaces # Library wajib untuk tier ZeroGPU
+import spaces
 
-# Load model (tambahkan compile=False agar aman)
-model = tf.keras.models.load_model('cat_model.h5', compile=False)
+# Load model dengan format .keras (format modern, bebas bug Flatten)
+model = tf.keras.models.load_model('cat_model.keras')
 
 classes = ["Abyssinian", "Bengal", "Bombay", "British Shorthair", "Domestic",
            "Maine Coon", "Persian", "Ragdoll", "Siamese", "Sphynx"]
 
-# Dekorator ini wajib ada agar fungsinya bisa jalan di ZeroGPU
 @spaces.GPU
 def predict_cat(image):
     if image is None: