Spaces:
Running
Running
File size: 2,093 Bytes
407df6a 3736885 2b1e0ff 407df6a c840d28 407df6a 60c015e 407df6a 8b60ac4 407df6a 5a07ece 407df6a 68ad70a 407df6a 84afdfc 407df6a |
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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
import gradio as gr
import requests
import concurrent.futures
import os
SECRET_TOKEN = os.getenv("KEY")
def prompt(pos,hei,wid,no):
hed={
"Content-Type": "application/json",
"Authorization": "Bearer " + SECRET_TOKEN
}
j={
"deployment_id": "2ce5158e-7fa0-4dd6-b533-550c1e8f8c3b",
"inputs": {
"input_text": f"{pos} (clear image: 1.1), (clean image: 1.1),masterpiece , best quality , hires , 8k, ((midlength)) ",
"Negative": "worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, artist name,text, watermark,(Low quality , blurred , bad hands ,diffused body, ugly face , worst quality, low quality, normal quality, painting, drawing, monochrome, grayscale, bad anatomy, bad proportions, signature, watermark)",
"Width ": f"{wid}",
"Height ": f"{hei}",
"Batch": "",
}
}
id = []
urls=[]
p=0
no=no
def gen():
r = requests.post("https://www.comfydeploy.com/api/run",headers=hed,json=j)
a=r.json()
a=a['run_id']
id.append(a)
futures = []
with concurrent.futures.ThreadPoolExecutor() as executor:
for url in range(no):
futures.append(executor.submit(gen))
for future in concurrent.futures.as_completed(futures):
print(future)
for i in id :
li = requests.get(f"https://www.comfydeploy.com/api/run?run_id={i}",headers=hed)
lo=li.json()
url=lo['outputs']
while url ==[]:
li = requests.get(f"https://www.comfydeploy.com/api/run?run_id={i}",headers=hed)
lo=li.json()
url =lo['outputs']
while 'url' not in url:
li = requests.get(f"https://www.comfydeploy.com/api/run?run_id={i}",headers=hed)
lo=li.json()
url=lo['outputs'][0]['data']['images'][0]
url=url['url']
urls.append(url)
return urls
with gr.Blocks() as iface:
gr.Interface(fn=prompt, inputs=["text",gr.Slider(8,1024,step=8.0 ,value=512),gr.Slider(8,1024,step=8.0, value=512),gr.Slider(1,10,step=1.0)],outputs="gallery")
if __name__ == "__main__":
iface.launch()
|