Update app.py

app.py

@@ -190,6 +190,11 @@ class BERTDataset(Dataset):
     def __len__(self):
         return (len(self.labels))
 
+
+tokenizer = KoBERTTokenizer.from_pretrained('skt/kobert-base-v1')
+bertmodel = BertModel.from_pretrained('skt/kobert-base-v1', return_dict=False)
+vocab = nlp.vocab.BERTVocab.from_sentencepiece(tokenizer.vocab_file, padding_token='[PAD]')
+
 # Kobert_softmax
 class BERTClassifier(nn.Module):
     def __init__(self,
@@ -614,16 +619,16 @@ with gr.Blocks() as app:
                 choices=[f"{k}. {v}" for k, v in options.items()],
                 label="🤖 원하는 챗봇 스타일 선택"
             )
-            diary_input = gr.Textbox(label="📜 오늘의 하루 기록하기", placeholder="ex)오늘 소풍가서 맛있는 걸 많이 먹어서 엄청 신났어")
-            image_input = gr.Image(type="pil", label="📷 얼굴 표정 사진 업로드")
             playlist_input = gr.Radio(["비슷한", "상반된"], label="🎧 오늘의 감정과 ㅇㅇ되는 플레이리스트 추천 받기")
+            diary_input = gr.Textbox(label="📜 오늘의 하루 기록하기", placeholder="ex)오늘 소풍가서 맛있는 걸 많이 먹어서 엄청 신났어")
+            image_input = gr.Image(type="pil", label="📷 얼굴 표정 사진 업로드", width=256, height=256)
             submit_btn = gr.Button("🚀 분석 시작")
 
         with gr.Column():
             output_playlist = gr.Dataframe(label="🎧 추천 플레이리스트 ")
             output_comment = gr.Textbox(label="💬 AI 코멘트")
             output_topics = gr.Textbox(label="📝 추천 일기 콘텐츠")
-            output_image = gr.Image(label="🖼️ 생성된 오늘의 감정 캐릭터", type="pil", width=512, height=512)
+            output_image = gr.Image(label="🖼️ 생성된 오늘의 감정 캐릭터", type="pil", width=256, height=256)
 
     # 버튼 클릭 이벤트 연결
     submit_btn.click(