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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import streamlit as st
7
  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, token=huggingface_token)
12
  model = AutoModelForCausalLM.from_pretrained(model_name, token=huggingface_token, torch_dtype=torch.float16)
13
 
 
7
  huggingface_token = st.secrets["HUGGINGFACEHUB_API_TOKEN"]
8
 
9
  # Cargar el modelo y tokenizer
10
+ model_name = "meta-llama/Llama-3.1-8B-Instruct"
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