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

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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
  # 학교 정보 및 일정 가져오기 함수