MaduRox commited on
Commit
bbd3bde
·
1 Parent(s): 9bf8aab

Fix: top-level import spaces and @spaces.GPU(duration=30) for HF ZeroGPU

Browse files
Files changed (1) hide show
  1. app.py +6 -8
app.py CHANGED
@@ -17,14 +17,12 @@ from fastapi import FastAPI, HTTPException, Request, Header, Depends, File, Uplo
17
  from fastapi.responses import JSONResponse, RedirectResponse, Response
18
  from fastapi.middleware.cors import CORSMiddleware
19
  from pydantic import BaseModel, Field
20
- try:
21
- import spaces
22
- @spaces.GPU
23
- def zero_gpu_keepalive():
24
- """Satisfies HF ZeroGPU startup check."""
25
- return "ZeroGPU active"
26
- except Exception as e:
27
- print(f"Notice: ZeroGPU spaces module not imported ({e}). Running in standard environment.")
28
 
29
  # ── RIF Engine (Pure CPU, O(1) Memory) ──────────────────────────
30
  class RIFEngine:
 
17
  from fastapi.responses import JSONResponse, RedirectResponse, Response
18
  from fastapi.middleware.cors import CORSMiddleware
19
  from pydantic import BaseModel, Field
20
+ import spaces
21
+
22
+ @spaces.GPU(duration=30)
23
+ def zero_gpu_keepalive(query: str = "keepalive"):
24
+ """ZeroGPU keepalive function required by HF ZeroGPU runtime."""
25
+ return f"ZeroGPU active: {query}"
 
 
26
 
27
  # ── RIF Engine (Pure CPU, O(1) Memory) ──────────────────────────
28
  class RIFEngine: