chobob311 commited on
Commit
81b24be
ยท
verified ยท
1 Parent(s): fefdb18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,9 +1,12 @@
1
  import gradio as gr
2
  from transformers import LlamaTokenizer, LlamaForCausalLM
3
  import torch
 
4
 
5
  model_id = 'Bllossom/llama-3-Korean-Bllossom-70B'
6
- hf_access_token = 'ใ…—ใ…—' # ์‹ค์ œ ํ—ˆ๊น…ํŽ˜์ด์Šค ์•ก์„ธ์Šค ํ† ํฐ์œผ๋กœ ๊ต์ฒดํ•˜์„ธ์š”
 
 
7
 
8
  # ํ† ํฌ๋‚˜์ด์ €์™€ ๋ชจ๋ธ ๋กœ๋“œ
9
  tokenizer = LlamaTokenizer.from_pretrained(
 
1
  import gradio as gr
2
  from transformers import LlamaTokenizer, LlamaForCausalLM
3
  import torch
4
+ import os # os ๋ชจ๋“ˆ ์ž„ํฌํŠธ
5
 
6
  model_id = 'Bllossom/llama-3-Korean-Bllossom-70B'
7
+
8
+ # ํ™˜๊ฒฝ ๋ณ€์ˆ˜์—์„œ ์•ก์„ธ์Šค ํ† ํฐ ๊ฐ€์ ธ์˜ค๊ธฐ
9
+ hf_access_token = os.getenv('HUGGINGFACEHUB_API_TOKEN')
10
 
11
  # ํ† ํฌ๋‚˜์ด์ €์™€ ๋ชจ๋ธ ๋กœ๋“œ
12
  tokenizer = LlamaTokenizer.from_pretrained(