matteomarjanovic commited on
Commit
5b3343a
·
1 Parent(s): 714fad0

add hf login

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -5,12 +5,15 @@ import random
5
  import spaces #[uncomment to use ZeroGPU]
6
  from diffusers import DiffusionPipeline
7
  from controlnet_aux import CannyDetector
 
8
  import torch
9
  import subprocess
10
  from groq import Groq
11
  import base64
12
  import os
13
 
 
 
14
  subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
15
 
16
  # Load FLUX image generator
 
5
  import spaces #[uncomment to use ZeroGPU]
6
  from diffusers import DiffusionPipeline
7
  from controlnet_aux import CannyDetector
8
+ from huggingface_hub import login
9
  import torch
10
  import subprocess
11
  from groq import Groq
12
  import base64
13
  import os
14
 
15
+ login(token=os.environ.get("HF_API_KEY"))
16
+
17
  subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
18
 
19
  # Load FLUX image generator