Update src/v2_for_hf.py

src/v2_for_hf.py

@@ -6,7 +6,10 @@ import re
 from diffusers import DiffusionPipeline
 import torch
 from PIL import Image
+
 import os
+from huggingface_hub import login
+login(token=os.environ.get("HF_token"))
 
 # Modfiy this to change the number of generations
 NUM_GEN = 2