Ron1006 commited on
Commit
ccc3a41
·
verified ·
1 Parent(s): 52f7d48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -1,7 +1,15 @@
1
- # -*- coding: utf-8 -*-
2
  import gradio as gr
3
  from PIL import Image
4
  from diffusers import DiffusionPipeline
 
 
 
 
 
 
 
 
5
 
6
  # 加载 Hugging Face 上的预训练模型
7
  pipe = DiffusionPipeline.from_pretrained("lamblabs/sd-image-variations-diffusers")
 
1
+ # -*- coding: utf-8 -*-
2
  import gradio as gr
3
  from PIL import Image
4
  from diffusers import DiffusionPipeline
5
+ from huggingface_hub import login
6
+ import os
7
+
8
+ # 从环境变量中获取访问令牌
9
+ hf_token = os.getenv("hf_token")
10
+
11
+ # 使用令牌进行登录
12
+ login(token=hf_token)
13
 
14
  # 加载 Hugging Face 上的预训练模型
15
  pipe = DiffusionPipeline.from_pretrained("lamblabs/sd-image-variations-diffusers")