Dagriffpatchfan commited on
Commit
71b8478
·
verified ·
1 Parent(s): d7512eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -1,12 +1,9 @@
1
  import gradio as gr
2
  from llama_cpp import Llama
3
  from huggingface_hub import login
4
- import contextlib
5
- import io
6
-
7
- with contextlib.redirect_stdout(io.StringIO()):
8
- login(token=hf_token)
9
-
10
 
11
  # Load the model (only once)
12
  llm = Llama.from_pretrained(
 
1
  import gradio as gr
2
  from llama_cpp import Llama
3
  from huggingface_hub import login
4
+ import os
5
+ my_hf_token = os.getenv("hf_token")
6
+ login(token=my_hf_token)
 
 
 
7
 
8
  # Load the model (only once)
9
  llm = Llama.from_pretrained(