Update main.py

main.py

@@ -73,4 +73,8 @@ async def run_gen_task(task_id: str, prompt: str, image_link: str):
     if res.get("ok"):
         tasks_status[task_id] = {"status": "success", "result": res["img_url"]}
     else:
-        tasks_status[task_id] = {"status": "failed", "error": res.get("error")}
+        tasks_status[task_id] = {"status": "failed", "error": res.get("error")}
+
+@app.get("/")
+async def root():
+    return {"status": "running"}