Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -47,7 +47,17 @@ class CustomDataset(Dataset):
|
|
| 47 |
# 훈련 데이터 준비 및 모델 훈련
|
| 48 |
def train_model():
|
| 49 |
csv_data = """description,gender
|
| 50 |
-
"그는 축구를 정말
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
"""
|
| 52 |
|
| 53 |
data = pd.read_csv(io.StringIO(csv_data))
|
|
|
|
| 47 |
# 훈련 데이터 준비 및 모델 훈련
|
| 48 |
def train_model():
|
| 49 |
csv_data = """description,gender
|
| 50 |
+
"그는 축구를 정말 좋아하고, 근육질의 몸매를 가졌다.",남자
|
| 51 |
+
"그녀는 긴 머리를 가졌고, 분홍색 원피스를 입었다.",여자
|
| 52 |
+
"짧은 머리에 정장을 입은 그는 회의에 참석했다.",남자
|
| 53 |
+
"아름다운 목소리로 노래하는 그녀는 가수다.",여자
|
| 54 |
+
"그의 취미는 자동차 정비와 컴퓨터 게임이다.",남자
|
| 55 |
+
"그녀는 섬세한 손길로 아기 인형을 만들었다.",여자
|
| 56 |
+
"군대에서 막 제대한 그는 씩씩해 보였다.",남자
|
| 57 |
+
"그녀는 친구들과 수다 떠는 것을 좋아한다.",여자
|
| 58 |
+
"강력한 리더십으로 팀을 이끄는 모습이 인상적이었다.",남자
|
| 59 |
+
"자신이 직접 만든 쿠키를 주변에 나누어주곤 한다.",여자
|
| 60 |
+
"정민지",천사사
|
| 61 |
"""
|
| 62 |
|
| 63 |
data = pd.read_csv(io.StringIO(csv_data))
|