vincentkyo commited on
Commit
5772e57
·
verified ·
1 Parent(s): e8c85aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -8,7 +8,7 @@ from huggingface_hub import InferenceClient
8
  from PIL import Image
9
 
10
 
11
- async def run_image_to_image(img: Image.Image, prompt: str, model: str) -> Image.Image:
12
  token = os.getenv("HF_TOKEN")
13
  if not token:
14
  raise RuntimeError("HF_TOKEN is not set. Please set your Hugging Face token in environment variables.")
@@ -23,7 +23,7 @@ async def run_image_to_image(img: Image.Image, prompt: str, model: str) -> Image
23
  input_bytes = f.read()
24
 
25
  try:
26
- result = await client.image_to_image(
27
  input_bytes,
28
  prompt=prompt,
29
  model=model,
@@ -79,8 +79,8 @@ with gr.Blocks(title="图像编辑工具", css=".gradio-container {max-width: 98
79
  - 将图片背景变成樱花盛开的场景
80
  """)
81
 
82
- async def _run_and_pack(img: Image.Image, p: str, m: str):
83
- result_img = await run_image_to_image(img, p, m)
84
  # 保存为临时文件以供下载
85
  with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
86
  result_img.save(tmp.name)
 
8
  from PIL import Image
9
 
10
 
11
+ def run_image_to_image(img: Image.Image, prompt: str, model: str) -> Image.Image:
12
  token = os.getenv("HF_TOKEN")
13
  if not token:
14
  raise RuntimeError("HF_TOKEN is not set. Please set your Hugging Face token in environment variables.")
 
23
  input_bytes = f.read()
24
 
25
  try:
26
+ result = client.image_to_image(
27
  input_bytes,
28
  prompt=prompt,
29
  model=model,
 
79
  - 将图片背景变成樱花盛开的场景
80
  """)
81
 
82
+ def _run_and_pack(img: Image.Image, p: str, m: str):
83
+ result_img = run_image_to_image(img, p, m)
84
  # 保存为临时文件以供下载
85
  with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
86
  result_img.save(tmp.name)