alessandro trinca tornidor commited on
Commit
05528cb
·
1 Parent(s): 8f287b5

feat: revert use of spaces.GPU on infer_lisa_gradio()

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -55,7 +55,7 @@ async def request_middleware(request, call_next):
55
 
56
  return await logging_middleware(request, call_next)
57
 
58
- #@spaces.GPU
59
  def gpu_initialization() -> None:
60
  app_logger.info("GPU initialization...")
61
 
@@ -120,7 +120,10 @@ async def health() -> JSONResponse:
120
  return JSONResponse(status_code=200, content={"msg": "still alive..."})
121
 
122
 
123
- # @spaces.GPU
 
 
 
124
  def infer_lisa_gradio(request_input: StringPromptApiRequestBody) -> str:
125
  from samgis_lisa.io_package.wrappers_helpers import get_parsed_bbox_points_with_string_prompt
126
  from samgis_lisa.prediction_api import lisa
 
55
 
56
  return await logging_middleware(request, call_next)
57
 
58
+ @spaces.GPU
59
  def gpu_initialization() -> None:
60
  app_logger.info("GPU initialization...")
61
 
 
120
  return JSONResponse(status_code=200, content={"msg": "still alive..."})
121
 
122
 
123
+ # try executing gpu_initialization() not within infer_lisa_gradio()
124
+ # gpu_initialization()
125
+
126
+ @spaces.GPU
127
  def infer_lisa_gradio(request_input: StringPromptApiRequestBody) -> str:
128
  from samgis_lisa.io_package.wrappers_helpers import get_parsed_bbox_points_with_string_prompt
129
  from samgis_lisa.prediction_api import lisa