Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +3 -1
src/streamlit_app.py
CHANGED
@@ -3,12 +3,14 @@ import requests
|
|
3 |
import streamlit as st
|
4 |
from datetime import datetime
|
5 |
from transformers import pipeline
|
|
|
|
|
|
|
6 |
|
7 |
# Gemma 모델 로딩
|
8 |
@st.cache_resource
|
9 |
def load_model():
|
10 |
return pipeline("text-generation", model="google/gemma-2-2b-it", max_new_tokens=512)
|
11 |
-
|
12 |
llm = load_model()
|
13 |
|
14 |
# 학교 정보 및 일정 가져오기 함수
|
|
|
3 |
import streamlit as st
|
4 |
from datetime import datetime
|
5 |
from transformers import pipeline
|
6 |
+
from huggingface_hub import login
|
7 |
+
login(token=os.environ.get("HUGGINGFACE_TOKEN"))
|
8 |
+
|
9 |
|
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 |
# 학교 정보 및 일정 가져오기 함수
|