Update app.py

app.py

@@ -2,11 +2,16 @@ import streamlit as st
 from transformers import pipeline
 
 # 모델 로드
-model = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")
+model = pipeline("text-generation", model="distilgpt2")  # 경량화 모델 사용
 
-# 웹 인터페이스 생성
-st.title("GPT-N Model")
-input_text = st.text_input("Enter your prompt:")
+# 앱 제목
+st.title("GPT Text Generator")
+
+# 사용자 입력
+prompt = st.text_input("Enter your prompt:")
+
+# 버튼 클릭 이벤트 감싸기
 if st.button("Generate"):
-    result = model(input_text)
-    st.write(result[0]["generated_text"])
+    with st.spinner("Generating..."):  # 로딩 스피너 추가
+        result = model(prompt, max_length=100)
+        st.write(result[0]["generated_text"])