SebastianItamari commited on
Commit
dad94cd
·
verified ·
1 Parent(s): f774133

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def login_to_huggingface():
15
  # Inicializar el modelo generador de texto
16
  @st.cache_resource # Cachear el modelo para no recargarlo repetidamente
17
  def load_model():
18
- model_id = "meta-llama/Llama-2-7b" # Cambia al modelo correcto disponible
19
  tokenizer = AutoTokenizer.from_pretrained(model_id)
20
  model = AutoModelForCausalLM.from_pretrained(
21
  model_id,
 
15
  # Inicializar el modelo generador de texto
16
  @st.cache_resource # Cachear el modelo para no recargarlo repetidamente
17
  def load_model():
18
+ model_id = "meta-llama/Llama-3.2-1B" # Cambia al modelo correcto disponible
19
  tokenizer = AutoTokenizer.from_pretrained(model_id)
20
  model = AutoModelForCausalLM.from_pretrained(
21
  model_id,