Huggingface入门
Huggingface:https://huggingface.co/
选取Models模块,搜搜flux,选择XLabs-AI/flux-RealismLora
在API部分选择View Code:

得到以下代码:
import requests
API_URL = "https://api-inference.huggingface.co/models/XLabs-AI/flux-RealismLora"
headers = {"Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
def query(payload):
response = requests.post(API_URL, headers=headers, json=payload)
return response.content
image_bytes = query({
"inputs": "Astronaut riding a horse",
})
# You can access the image with PIL.Image for example
import io
from PIL import Image
image = Image.open(io.BytesIO(image_bytes))
选择tokens:

创建新token

选择write,填入自定义的token name

把create的token替换代码中的hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx,再加工一下:
import gradio as gr
import requests
import io
from PIL import Image
API_URL = "https://api-inference.huggingface.co/models/XLabs-AI/flux-RealismLora"
headers = {"Authorization": "Bearer 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"}
def query(payload):
response = requests.post(API_URL, headers=headers, json=payload)
return response.content
def generate_image(prompt):
image_bytes = query({"inputs": prompt})
image = Image.open(io.BytesIO(image_bytes))
return image
# 创建 Gradio 界面
iface = gr.Interface(
fn=generate_image,
inputs=gr.Textbox(label="输入提示词", placeholder="描述你的图像...(最好是英文)"),
outputs=gr.Image(label="生成的图像"),
title="Flux 图像生成 by FAVE",
description="输入一个描述,生成相应的图像。"
)
# 启动 Gradio 应用
iface.launch(share=True)
即可生成一个可分享给他人的链接,打开是一个图形化界面的图片生成网站

我利用"Two anime characters, a man and a woman, sitting on a bench at the foot of Mount Fuji, back to back"生成了以下图片:


浙公网安备 33010602011771号