alexeyGod commited on
Commit
f058d4e
·
verified ·
1 Parent(s): 506c05b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -2,10 +2,15 @@ import gradio as gr
2
  import numpy as np
3
  import random
4
 
5
- # import spaces #[uncomment to use ZeroGPU]
6
  from diffusers import DiffusionPipeline
7
  import torch
8
 
 
 
 
 
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
  model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like to use
11
 
 
2
  import numpy as np
3
  import random
4
 
5
+ import spaces #[uncomment to use ZeroGPU]
6
  from diffusers import DiffusionPipeline
7
  import torch
8
 
9
+ from huggingface_hub import login
10
+ import os
11
+ a=os.getenv('hf_tok')
12
+ login(token=a )
13
+
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
  model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like to use
16