manejo de secrets para el modelo

.env-example

@@ -0,0 +1,2 @@
+HF_AUTH_TOKEN=your_huggingface_token
+API_KEY=your_api_key

.gitignore

@@ -0,0 +1,3 @@
+venv/
+.env
+test.py

app.py

@@ -1,7 +1,18 @@
 import streamlit as st
 from utils import carga_modelo, genera
+from dotenv import load_dotenv
 import os
 
+# Load environment variables from .env
+load_dotenv()
+
+# Access the variables
+token = os.getenv("HF_AUTH_TOKEN")
+api_key = os.getenv("API_KEY")
+repo_id = 'ceyda/butterfly_cropped_uniq1K_512'
+modelo_gan = carga_modelo(repo_id, token)
+
+
 # Página principal
 st.title('Generador de Mariposas')
 st.write('Este es un modelo light GAN entrenado para generación de mariposas')

requirements.txt

@@ -1,5 +1,6 @@
 git+https://github.com/huggingface/community-events.git@3fea10c5d5a50c69f509e34cd580fe9139905d04#egg=huggan
 transformers
+python-dotenv
 
 # faiss-cpu
 # paddlehub

utils.py

@@ -1,13 +1,18 @@
 import numpy as np
 import torch 
 from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
+from transformers import AutoModel
 
 
 ## Cargamos el modelo desde el Hub de Hugging Face
-def carga_modelo(model_name='ceyda/butterfly_cropped_uniq1K_512', model_version=None):
-    gan = LightweightGAN.from_pretrained(model_name, version=model_version)
-    gan.eval()
-    return gan
+def carga_modelo(repo_id, token):
+    return AutoModel.from_pretrained(repo_id, use_auth_token=token)
+
+
+# def carga_modelo(model_name='ceyda/butterfly_cropped_uniq1K_512', model_version=None):
+#     gan = LightweightGAN.from_pretrained(model_name, version=model_version)
+#     gan.eval()
+#     return gan
 
 def genera(gan, batch_size=1):
     with torch.no_grad():