diff --git a/main.py b/main.py
index 32b6003..7b4c8c6 100644
--- a/main.py
+++ b/main.py
@@ -21,7 +21,7 @@ transforms = T.Compose([
 
 task = Task.init(
     project_name="Person_Attribute_Recognition", 
-    task_name="MarketDuke_Agent"
+    task_name="model-yolo-person-classify"
 )