Smartlizardpy commited on
Commit
e5a5a0d
·
verified ·
1 Parent(s): 26e6a51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,9 +1,10 @@
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  from huggingface_hub import login
 
4
 
5
  # Authenticate using token
6
- login(token="your_token_here")
7
 
8
  # Load the model and tokenizer
9
  model_name = "mistralai/Mistral-Nemo-Instruct-2407"
 
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  from huggingface_hub import login
4
+ import os
5
 
6
  # Authenticate using token
7
+ login(token=os.getenv('hf_token'))
8
 
9
  # Load the model and tokenizer
10
  model_name = "mistralai/Mistral-Nemo-Instruct-2407"