JaphetHernandez commited on
Commit
438e2df
·
verified ·
1 Parent(s): b192930

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ huggingface_token = st.secrets["HUGGINGFACEHUB_API_TOKEN"]
8
 
9
  # Cargar el modelo y tokenizer
10
  model_name = "meta-llama/Llama-3.1-8B"
11
- tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=huggingface_token)
12
- model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=huggingface_token, torch_dtype=torch.float16)
13
 
14
  # Usar Streamlit para cargar el archivo CSV
15
  uploaded_file = st.file_uploader("Por favor sube un archivo CSV:", type="csv")
 
8
 
9
  # Cargar el modelo y tokenizer
10
  model_name = "meta-llama/Llama-3.1-8B"
11
+ tokenizer = AutoTokenizer.from_pretrained(model_name, token=huggingface_token)
12
+ model = AutoModelForCausalLM.from_pretrained(model_name, token=huggingface_token, torch_dtype=torch.float16)
13
 
14
  # Usar Streamlit para cargar el archivo CSV
15
  uploaded_file = st.file_uploader("Por favor sube un archivo CSV:", type="csv")