Update app.py

app.py

@@ -1,4 +1,7 @@
 import os
+os.environ["KERAS_BACKEND"] = "jax"
+os.environ["JAX_PLATFORM_NAME"] = "cpu"  # Fast startup, no GPU needed
+
 import base64
 import logging
 import numpy as np
@@ -6,9 +9,6 @@ import cv2
 import requests
 from typing import List, Dict, Any
 
-os.environ["KERAS_BACKEND"] = "jax"
-os.environ["JAX_PLATFORM_NAME"] = "cpu"  # Fast startup, no GPU needed
-
 import keras
 from fastapi import FastAPI, File, UploadFile, HTTPException
 from huggingface_hub import hf_hub_download