Update app.py

app.py

@@ -1,16 +1,12 @@
 import streamlit as st
 from transformers import pipeline
 
-# 제목
-st.title("My GPT Web App")
+# 모델 로드
+model = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")
 
-# 사용자 입력
-user_input = st.text_input("Enter your prompt:")
-
-# 모델 로드 (PyTorch 사용)
-gpt_model = pipeline("text-generation", model="gpt2", framework="pt")
-
-# 버튼 클릭 시 결과 출력
+# 웹 인터페이스 생성
+st.title("GPT-N Model")
+input_text = st.text_input("Enter your prompt:")
 if st.button("Generate"):
-    response = gpt_model(user_input, max_length=100)
-    st.write(response[0]['generated_text'])
+    result = model(input_text)
+    st.write(result[0]["generated_text"])