yongyeol commited on
Commit
e14a8ca
·
verified ·
1 Parent(s): ec2e83b

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +6 -1
src/streamlit_app.py CHANGED
@@ -10,7 +10,12 @@ login(token=os.environ.get("HUGGINGFACE_TOKEN"))
10
  # Gemma 모델 로딩
11
  @st.cache_resource
12
  def load_model():
13
- return pipeline("text-generation", model="google/gemma-2-2b-it", max_new_tokens=512)
 
 
 
 
 
14
  llm = load_model()
15
 
16
  # 학교 정보 및 일정 가져오기 함수
 
10
  # Gemma 모델 로딩
11
  @st.cache_resource
12
  def load_model():
13
+ return pipeline(
14
+ "text-generation",
15
+ model="google/gemma-2-2b-it",
16
+ max_new_tokens=512,
17
+ use_auth_token=os.environ.get("HUGGINGFACE_TOKEN") # ✅ 직접 전달
18
+ )
19
  llm = load_model()
20
 
21
  # 학교 정보 및 일정 가져오기 함수