Update app.py
Browse files
app.py
CHANGED
|
@@ -6,6 +6,9 @@ import cv2
|
|
| 6 |
import requests
|
| 7 |
from typing import List, Dict, Any
|
| 8 |
|
|
|
|
|
|
|
|
|
|
| 9 |
import keras
|
| 10 |
from fastapi import FastAPI, File, UploadFile, HTTPException
|
| 11 |
from huggingface_hub import hf_hub_download
|
|
@@ -13,8 +16,7 @@ from huggingface_hub import hf_hub_download
|
|
| 13 |
# βββ Logging & Backend ββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 14 |
logging.basicConfig(level=logging.INFO)
|
| 15 |
logger = logging.getLogger(__name__)
|
| 16 |
-
|
| 17 |
-
os.environ["JAX_PLATFORM_NAME"] = "cpu" # Fast startup, no GPU needed
|
| 18 |
|
| 19 |
# βββ Config (use Space Secrets for API_KEY) βββββββββββββββββββββββββββββββββββ
|
| 20 |
API_KEY = os.getenv("ROBOFLOW_API_KEY") # Set in Space Settings β Secrets
|
|
|
|
| 6 |
import requests
|
| 7 |
from typing import List, Dict, Any
|
| 8 |
|
| 9 |
+
os.environ["KERAS_BACKEND"] = "jax"
|
| 10 |
+
os.environ["JAX_PLATFORM_NAME"] = "cpu" # Fast startup, no GPU needed
|
| 11 |
+
|
| 12 |
import keras
|
| 13 |
from fastapi import FastAPI, File, UploadFile, HTTPException
|
| 14 |
from huggingface_hub import hf_hub_download
|
|
|
|
| 16 |
# βββ Logging & Backend ββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 17 |
logging.basicConfig(level=logging.INFO)
|
| 18 |
logger = logging.getLogger(__name__)
|
| 19 |
+
|
|
|
|
| 20 |
|
| 21 |
# βββ Config (use Space Secrets for API_KEY) βββββββββββββββββββββββββββββββββββ
|
| 22 |
API_KEY = os.getenv("ROBOFLOW_API_KEY") # Set in Space Settings β Secrets
|