more stuff

app.py

@@ -1,7 +1,10 @@
 import gradio as gr
 import torch
 from PIL import Image
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, login
+import os
+
+login(token=os.environ["HUGGINGFACE_TOKEN"])
 
 demo_imgs = [
     ["images/chinchilla_web-1024x683.jpg", "images/shiba-inu-dog-in-the-snow.jpg"],
@@ -59,7 +62,6 @@ model.load_state_dict(torch.load(checkpoint_path), strict=False)
 
 model.eval()
 
-
 def generate(
     idx,
     image,