Update app.py

app.py

@@ -155,7 +155,7 @@ def update_task_status(category, task_id, status, result=None):
         print(f"Failed to update task {task_id}: {response.status_code} - {response.text}")
 
 
-@spaces.GPU(duration=200)
+@spaces.GPU(duration=150)
 def infer(request: gr.Request):
     if request:
         print("请求头字典:", request.headers)
@@ -202,11 +202,12 @@ def infer(request: gr.Request):
 
                 generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
                 parsed_answer = processor.post_process_generation(generated_text, task=prompt, image_size=(image.width, image.height))
+                print(task['id'])
+                print(parsed_answer)
 
                 # Update the task status to Successed with result
                 update_task_status('img2text', task['id'], 'Successed', {"text": parsed_answer})
 
-                return parsed_answer
             except Exception as e:
                 print(f"Error processing task {task['id']}: {e}")
                 # If error occurs, update the task status to Failed