Update app.py

app.py

@@ -207,6 +207,11 @@ def infer(request: gr.Request):
 
             # Update the task status to Successed with result
             update_task_status('img2text', task['id'], 'Successed', {"text": parsed_answer})
+        except Exception as e:
+            print(f"Error processing task {task['id']}: {e}")
+            # If error occurs, update the task status to Failed
+            update_task_status('img2text', task['id'], 'Failed')
+            return f"Error processing task {task['id']}: {e}"
 
     return "No pending tasks found."