| import gradio as gr | |
| import io | |
| from PIL import Image | |
| # gr.Interface.load("models/runwayml/stable-diffusion-v1-5").launch() | |
| import requests | |
| API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5" | |
| headers = {"Authorization": "Bearer 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"} | |
| def query(payload): | |
| response = requests.post(API_URL, headers=headers, json=payload) | |
| image = Image.open(io.BytesIO(response.content)) | |
| return image | |
| # 创建Gradio界面 | |
| iface = gr.Interface( | |
| fn=query, | |
| inputs="text", | |
| outputs="image", | |
| # live=True, | |
| title="text to image" | |
| ) | |
| # 启动Gradio应用 | |
| iface.launch() |