Update app.py

app.py

@@ -1,19 +1,12 @@
 import gradio as gr
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
-
-# 모델과 토크나이저를 전역 변수로 미리 로드
-model = GPT2LMHeadModel.from_pretrained("gpt2")
-tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
+from transformers import pipeline
 
 def generate_diary(keywords):
-    # 키워드 기반 fine-tuning
-    prompt = f"오늘의 일기:\n\n{' '.join(keywords.split(','))}에 관한 일기를 써봅시다."
-    input_ids = tokenizer.encode(prompt, return_tensors="pt")
-    output = model.generate(input_ids, max_length=500, num_return_sequences=1, do_sample=True, top_k=50, top_p=0.95, num_beams=5, no_repeat_ngram_size=2)
-    
-    # 생성된 일기 텍스트 반환
-    diary = tokenizer.decode(output[0], skip_special_tokens=True)
-    return diary
+    # InstructGPT 모델 활용
+    generator = pipeline('text-generation', model='anthropic/InstructGPT')
+    prompt = f"오늘의 일기:\n\n{', '.join(keywords.split(','))}에 대한 일기를 써봅시다."
+    output = generator(prompt, max_length=500, num_return_sequences=1, do_sample=True, top_k=50, top_p=0.95, num_beams=5, no_repeat_ngram_size=2)[0]['generated_text']
+    return output
 
 def app():
     with gr.Blocks() as demo: