gbrabbit commited on
Commit
2100340
ยท
1 Parent(s): 8f7bcc7

Add .env file support and detailed debugging

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -6,6 +6,16 @@ import traceback
6
  from transformers import AutoTokenizer, AutoModelForCausalLM
7
  import torch
8
 
 
 
 
 
 
 
 
 
 
 
9
  # ํ™˜๊ฒฝ ๋ณ€์ˆ˜์—์„œ๋งŒ ํ† ํฐ ๊ฐ€์ ธ์˜ค๊ธฐ (๋ณด์•ˆ)
10
  HF_TOKEN = os.getenv("HF_TOKEN")
11
  MODEL_NAME = os.getenv("MODEL_NAME", "gbrabbit/lily-math-model")
@@ -186,7 +196,7 @@ with gr.Blocks(title="Lily Math RAG System", theme=gr.themes.Soft()) as demo:
186
  gr.Markdown(f"**๋ชจ๋ธ**: {MODEL_NAME}")
187
  gr.Markdown(f"**๋ชจ๋ธ ์ƒํƒœ**: {'โœ… ๋กœ๋“œ๋จ' if MODEL_LOADED else 'โŒ ๋กœ๋“œ ์‹คํŒจ'}")
188
  gr.Markdown(f"**ํ† ํฐ ์ƒํƒœ**: {'โœ… ์„ค์ •๋จ' if HF_TOKEN else 'โŒ ์„ค์ •๋˜์ง€ ์•Š์Œ'}")
189
- gr.Markdown("**๋ฒ„์ „**: 2.7.0 (์ƒ์„ธ ๋””๋ฒ„๊น…)")
190
 
191
  if __name__ == "__main__":
192
  demo.launch()
 
6
  from transformers import AutoTokenizer, AutoModelForCausalLM
7
  import torch
8
 
9
+ # .env ํŒŒ์ผ์—์„œ ํ™˜๊ฒฝ ๋ณ€์ˆ˜ ๋กœ๋“œ
10
+ try:
11
+ from dotenv import load_dotenv
12
+ load_dotenv()
13
+ print("โœ… .env ํŒŒ์ผ ๋กœ๋“œ๋จ")
14
+ except ImportError:
15
+ print("โš ๏ธ python-dotenv๊ฐ€ ์„ค์น˜๋˜์ง€ ์•Š์Œ, ์‹œ์Šคํ…œ ํ™˜๊ฒฝ ๋ณ€์ˆ˜ ์‚ฌ์šฉ")
16
+ except Exception as e:
17
+ print(f"โš ๏ธ .env ํŒŒ์ผ ๋กœ๋“œ ์‹คํŒจ: {e}")
18
+
19
  # ํ™˜๊ฒฝ ๋ณ€์ˆ˜์—์„œ๋งŒ ํ† ํฐ ๊ฐ€์ ธ์˜ค๊ธฐ (๋ณด์•ˆ)
20
  HF_TOKEN = os.getenv("HF_TOKEN")
21
  MODEL_NAME = os.getenv("MODEL_NAME", "gbrabbit/lily-math-model")
 
196
  gr.Markdown(f"**๋ชจ๋ธ**: {MODEL_NAME}")
197
  gr.Markdown(f"**๋ชจ๋ธ ์ƒํƒœ**: {'โœ… ๋กœ๋“œ๋จ' if MODEL_LOADED else 'โŒ ๋กœ๋“œ ์‹คํŒจ'}")
198
  gr.Markdown(f"**ํ† ํฐ ์ƒํƒœ**: {'โœ… ์„ค์ •๋จ' if HF_TOKEN else 'โŒ ์„ค์ •๋˜์ง€ ์•Š์Œ'}")
199
+ gr.Markdown("**๋ฒ„์ „**: 2.8.0 (.env ํŒŒ์ผ ์ง€์›)")
200
 
201
  if __name__ == "__main__":
202
  demo.launch()