File size: 1,397 Bytes
bc87293 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
import gradio as gr
import requests
def generate_image(prompt, negative_prompt, sampler, seed, guidance_scale, inference_steps):
invoke_url = "https://api.nvcf.nvidia.com/v2/nvcf/pexec/functions/89848fb8-549f-41bb-88cb-95d6597044a4"
fetch_url_format = "https://api.nvcf.nvidia.com/v2/nvcf/pexec/status/"
headers = {
"Authorization": "Bearer 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",
"Accept": "application/json",
}
payload = {
"prompt": prompt,
"negative_prompt": negative_prompt,
"sampler": sampler,
"seed": seed,
"guidance_scale": guidance_scale,
"inference_steps": inference_steps
}
session = requests.Session()
response = session.post(invoke_url, headers=headers, json=payload)
while response.status_code == 202:
request_id = response.headers.get("NVCF-REQID")
fetch_url = fetch_url_format + request_id
response = session.get(fetch_url, headers=headers)
response.raise_for_status()
response_body = response.json()
return response_body
iface = gr.Interface(fn=generate_image,
inputs=["text", "text", "text", "number", "number", "number"],
outputs="text",
title="Image Generation",
description="Generate an image based on prompts.")
iface.launch()
|