SebastianItamari commited on
Commit
aff9c25
·
verified ·
1 Parent(s): f869016

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,10 +1,15 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
  import torch
 
 
4
 
5
  # Inicializar el modelo generador de texto
6
  @st.cache_resource # Cachear el modelo para no recargarlo repetidamente
7
  def load_model():
 
 
 
8
  model_id = "meta-llama/Llama-2-7b" # Cambia al modelo correcto disponible
9
  return pipeline(
10
  "text-generation",
 
1
  import streamlit as st
2
  from transformers import pipeline
3
  import torch
4
+ import os
5
+ from huggingface_hub import login
6
 
7
  # Inicializar el modelo generador de texto
8
  @st.cache_resource # Cachear el modelo para no recargarlo repetidamente
9
  def load_model():
10
+ read_key = os.getenv("HUGGINGFACE_TOKEN")
11
+ login(token=read_key)
12
+
13
  model_id = "meta-llama/Llama-2-7b" # Cambia al modelo correcto disponible
14
  return pipeline(
15
  "text-generation",