JUNGU commited on
Commit
15ade12
Β·
verified Β·
1 Parent(s): 73b711a

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +8 -3
src/streamlit_app.py CHANGED
@@ -50,21 +50,26 @@ if os.getenv('OPENAI_API_KEY'):
50
  elif 'OPENAI_API_KEY' in st.secrets:
51
  st.session_state.openai_api_key = st.secrets['OPENAI_API_KEY']
52
 
 
 
 
 
 
53
  # ν•„μš”ν•œ NLTK 데이터 λ‹€μš΄λ‘œλ“œ
54
  try:
55
  nltk.data.find('tokenizers/punkt')
56
  except LookupError:
57
- nltk.download('punkt')
58
 
59
  try:
60
  nltk.data.find('tokenizers/punkt_tab')
61
  except LookupError:
62
- nltk.download('punkt_tab')
63
 
64
  try:
65
  nltk.data.find('corpora/stopwords')
66
  except LookupError:
67
- nltk.download('stopwords')
68
 
69
  # OpenAI API ν‚€ μ„€μ • (μ‹€μ œ μ‚¬μš© μ‹œ ν™˜κ²½ λ³€μˆ˜λ‚˜ Streamlit secretsμ—μ„œ κ°€μ Έμ˜€λŠ” 것이 μ’‹μŠ΅λ‹ˆλ‹€)
70
  if 'OPENAI_API_KEY' in os.environ:
 
50
  elif 'OPENAI_API_KEY' in st.secrets:
51
  st.session_state.openai_api_key = st.secrets['OPENAI_API_KEY']
52
 
53
+ # μž„μ‹œ 디렉토리λ₯Ό μ‚¬μš©ν•˜λ„λ‘ NLTK 데이터 경둜 μ„€μ •
54
+ nltk_data_dir = '/tmp/nltk_data'
55
+ os.makedirs(nltk_data_dir, exist_ok=True)
56
+ nltk.data.path.insert(0, nltk_data_dir) # 이 경둜λ₯Ό μš°μ„  κ²€μƒ‰ν•˜λ„λ‘ μ„€μ •
57
+
58
  # ν•„μš”ν•œ NLTK 데이터 λ‹€μš΄λ‘œλ“œ
59
  try:
60
  nltk.data.find('tokenizers/punkt')
61
  except LookupError:
62
+ nltk.download('punkt', download_dir=nltk_data_dir)
63
 
64
  try:
65
  nltk.data.find('tokenizers/punkt_tab')
66
  except LookupError:
67
+ nltk.download('punkt_tab', download_dir=nltk_data_dir)
68
 
69
  try:
70
  nltk.data.find('corpora/stopwords')
71
  except LookupError:
72
+ nltk.download('stopwords', download_dir=nltk_data_dir)
73
 
74
  # OpenAI API ν‚€ μ„€μ • (μ‹€μ œ μ‚¬μš© μ‹œ ν™˜κ²½ λ³€μˆ˜λ‚˜ Streamlit secretsμ—μ„œ κ°€μ Έμ˜€λŠ” 것이 μ’‹μŠ΅λ‹ˆλ‹€)
75
  if 'OPENAI_API_KEY' in os.environ: