Germano Cavalcante commited on
Commit
086be5c
1 Parent(s): 15d912e

Fix response_class in gpu_checker

Browse files
Files changed (1) hide show
  1. routers/tool_gpu_checker.py +2 -2
routers/tool_gpu_checker.py CHANGED
@@ -200,8 +200,8 @@ If that doesn't help, you can use Blender 2.79: https://www.blender.org/download
200
  return message
201
 
202
 
203
- @router.get("/gpu_checker")
204
- def gpu_checker(gpu_info: str = "", response_class=PlainTextResponse):
205
  message = gpu_checker_get_message(gpu_info)
206
 
207
  return message
 
200
  return message
201
 
202
 
203
+ @router.get("/gpu_checker", response_class=PlainTextResponse)
204
+ def gpu_checker(gpu_info: str = ""):
205
  message = gpu_checker_get_message(gpu_info)
206
 
207
  return message