Update app.py

app.py

@@ -7,8 +7,12 @@ from scipy import ndimage
 from tensorflow.keras import layers, models
 
 MODEL_PATH = "mnist_model.keras"
+import spaces  # استيراد مكتبة ZeroGPU
 
 
+# إضافة المزخرف فوق دالة التنبؤ
+@spaces.GPU
+
 def get_or_build_model():
     # 1. إذا كان النموذج موجوداً، قم بتحميله فوراً
     if os.path.exists(MODEL_PATH):
@@ -138,4 +142,5 @@ with gr.Blocks(title="MNIST Predictor") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo = gr.Interface(fn=predict_digit, inputs="sketchpad", outputs="label")
+demo.launch()