access token added

app.py

@@ -1,10 +1,18 @@
 import streamlit as st
+import os
 
 from utils import load_model, generate
 
 st.title('Image Generator with GAN')
 st.markdown('Butterflies generator')
 
+st.write("Token Access:", st.secrets["access_token"])
+
+st.write(
+	"Has environment variables been set:",
+	os.environ["access_token"] == st.secrets["access_token"])
+
+
 # Sidebar
 st.sidebar.subheader('This butterfly does not exist, it is completely generated!')
 st.sidebar.image('assets/logo.jpeg', width=200)

utils.py

@@ -1,10 +1,10 @@
 import numpy as np
-import torch 
+import torch
+import os
 from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
 
-
 def load_model(model_name='ceyda/butterfly_cropped_uniq1K_512', model_version=None):
-    gan = LightweightGAN.from_pretrained(model_name, version=model_version)
+    gan = LightweightGAN.from_pretrained(model_name, version=model_version, token=os.environ["access_token"])
     gan.eval()
     return gan