sairamv1 commited on
Commit
df82b88
Β·
verified Β·
1 Parent(s): 73d589b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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
- os.environ["KERAS_BACKEND"] = "jax"
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