Switch to tensorflow-cpu for faster loading on free tier

Dockerfile

@@ -15,7 +15,7 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir \
     huggingface_hub==0.20.3 \
     gradio==4.36.0 \
-    tensorflow==2.17.0 \
+    tensorflow-cpu==2.17.0 \
     numpy>=1.24.0 \
     Pillow>=10.0.0
 

app.py

@@ -4,9 +4,9 @@ import numpy as np
 from PIL import Image
 
 # Load model
-print("Loading DeepGuard model...")
+print("Loading DeepGuard model...", flush=True)
 model = tf.keras.models.load_model("deepfake_model.keras")
-print("Model loaded successfully!")
+print("Model loaded successfully!", flush=True)
 
 
 def generate_fft(image):