Update app.py

app.py

@@ -1418,6 +1418,14 @@ def predict_fraud_enhanced(file, threshold: float = 0.5):
         )
 
 
+# Apply GPU decorator to prediction function for ZeroGPU
+# This allows GPU to be allocated only during prediction calls, not for the entire server
+# ZeroGPU requires at least one @spaces.GPU decorated function to detect and allocate GPU
+if SPACES_AVAILABLE:
+    print("Applying @spaces.GPU decorator to predict_fraud_enhanced for ZeroGPU support")
+    predict_fraud_enhanced = spaces.GPU(predict_fraud_enhanced)
+
+
 # Create Gradio Interface
 with gr.Blocks(
     title="Credit Card Fraud Detection",