diff --git a/AI/model.py b/AI/model.py
index 7cef36ea..97ea7f9b 100644
--- a/AI/model.py
+++ b/AI/model.py
@@ -35,9 +35,8 @@ class Item_and_num(BaseModel) :
 @sbert.post('/team-building')
 async def start_sbert(data:Item_and_num):
     
-    input = data.items
-    min_num = data.teamMin
-    max_num = data.teamMax
+    input = data.participantResponses
+    max_num = data.maxTeamSize
     
 
     # 문장 임베딩