bkoz commited on
Commit
867a375
1 Parent(s): 45066df
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -9,6 +9,9 @@ from transformers import pipeline
9
  import gradio as gr
10
 
11
  token = os.getenv("HUGGINGFACE_API_TOKEN")
 
 
 
12
  pipe = pipeline("text-generation", "meta-llama/Meta-Llama-3-8B-Instruct", torch_dtype=torch.bfloat16, device_map="auto", token=token)
13
 
14
  # pipe.to('cuda')
 
9
  import gradio as gr
10
 
11
  token = os.getenv("HUGGINGFACE_API_TOKEN")
12
+
13
+ print(f'HUGGINGFACE_API_TOKEN {token}')
14
+
15
  pipe = pipeline("text-generation", "meta-llama/Meta-Llama-3-8B-Instruct", torch_dtype=torch.bfloat16, device_map="auto", token=token)
16
 
17
  # pipe.to('cuda')