nateraw commited on
Commit
763dae6
·
1 Parent(s): 0d9a07f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -6,6 +6,11 @@ from stable_diffusion_videos import StableDiffusionWalkPipeline, generate_images
6
  from diffusers.schedulers import LMSDiscreteScheduler
7
  import torch
8
 
 
 
 
 
 
9
 
10
  class ImageGenerationInterface:
11
  def __init__(self, pipeline):
 
6
  from diffusers.schedulers import LMSDiscreteScheduler
7
  import torch
8
 
9
+ import os
10
+ from huggingface_hub import HfFolder
11
+
12
+ HfFolder().set_token(os.environ['HF_TOKEN'])
13
+
14
 
15
  class ImageGenerationInterface:
16
  def __init__(self, pipeline):