Update space

Text2Long_text.py

@@ -9,7 +9,7 @@ tokenizer = AutoTokenizer.from_pretrained("skt/kogpt2-base-v2")
 model = AutoModelForCausalLM.from_pretrained("skt/kogpt2-base-v2").to(device)
 
 # 3. 한국어 소설 생성 함수
-def generate_korean_story(prompt, max_length=300):
+def generate_korean_story(prompt, max_length=100):
     input_ids = tokenizer.encode(prompt, return_tensors="pt").to(device)
 
     outputs = model.generate(
@@ -31,7 +31,7 @@ def generate_korean_story(prompt, max_length=300):
 # 4. 실행
 if __name__ == "__main__":
     user_prompt = input("📜 소설의 시작 문장을 입력하세요 (한국어): ")
-    result = generate_korean_story(user_prompt, max_length=500)
+    result = generate_korean_story(user_prompt, max_length=100)
     
     print("\n📖 생성된 한국어 소설:\n")
     print(result)