Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- 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(
|
|
|
|
|
|
|
|
|
|
|
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 |
# 학교 정보 및 일정 가져오기 함수
|