|
@@ -156,7 +156,7 @@ def get_image(
|
|
|
add_sampling_metadata=True,
|
|
|
api_name="/generate"
|
|
|
)
|
|
|
- background_tasks.add_task(remove_file, result)
|
|
|
+ background_tasks.add_task(remove_file, result[0])
|
|
|
return FileResponse(result, background=background_tasks)
|
|
|
|
|
|
@router.websocket("/sr")
|