Commit 9407f173 authored by siutin's avatar siutin
Browse files

store the last generated result

parent dbca5121

modules/call_queue.py

+1 −0
Original line number Diff line number Diff line
@@ -37,6 +37,7 @@ def wrap_gradio_gpu_call(func, extra_outputs=None): 
                res = func(*args, **kwargs)

            finally:

                progress.finish_task(id_task)

                progress.set_last_task_result(id_task, res)



            shared.state.end()

modules/progress.py

+10 −0
Original line number Diff line number Diff line
@@ -37,6 +37,16 @@ def finish_task(id_task): 
def add_task_to_queue(id_job):

    pending_tasks[id_job] = time.time()



last_task_id = None

last_task_result = None



def set_last_task_result(id_job, result):

  global last_task_id

  global last_task_result



  last_task_id = id_job

  last_task_result = result





class CurrentTaskResponse(BaseModel):

  current_task: str = Field(default=None, title="Task ID", description="id of the current progress task")

CRA Git | Maintained and supported by SUSTech CRA and CCSE