feat: 모델 변수명 및 requirement 수정

jiucchu · Jan 17, 2024 · fdf993d · fdf993d
1 parent 51ec4f2
commit fdf993d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/AI/model.py b/AI/model.py
@@ -29,16 +29,16 @@ class Item(BaseModel) :
     sentence : str
 
 class Item_and_num(BaseModel) :
-    item_list : List[Item]
-    min_n : int
-    max_n : int
+    items : List[Item]
+    teamMin : int
+    teamMax : int
 
 @sbert.post('/clustering')
 async def start_sbert(data:Item_and_num):
 
-    input = data.item_list
-    min_num = data.min_n
-    max_num = data.max_n
+    input = data.items
+    min_num = data.teamMin
+    max_num = data.teamMax
 
 
     # 문장 임베딩

diff --git a/AI/requirements.txt b/AI/requirements.txt