access token added

app.py

@@ -8,10 +8,6 @@ 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!')

utils.py

@@ -4,7 +4,7 @@ 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, use_auth_token=False)
+    gan = LightweightGAN.from_pretrained(model_name, version=model_version, use_auth_token=os.environ["access_token"])
     gan.eval()
     return gan