You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
27 lines
827 B
27 lines
827 B
7 months ago
|
from gradio_client import Client
|
||
|
|
|
||
|
|
# Пользователь - сессия
|
||
|
|
img_ses = {}
|
||
|
|
|
||
|
|
def draw(prompt, id):
|
||
|
|
global img_ses
|
||
|
|
if str(id) not in img_ses:
|
||
|
|
client = Client("https://playgroundai-playground-v2-5.hf.space/--replicas/0czym/")
|
||
|
|
img_ses[str(id)] = client
|
||
|
|
else:
|
||
|
|
client = img_ses[str(id)]
|
||
|
|
|
||
|
|
result = client.predict(
|
||
|
|
prompt, # Что рисуем
|
||
|
|
"", # 'Negative prompt'
|
||
|
|
False, # 'Use negative prompt'
|
||
|
|
0, # Сид
|
||
|
|
512, # float (numeric value between 256 and 1536) in 'Width' Slider component
|
||
|
|
512, # float (numeric value between 256 and 1536) in 'Height' Slider component
|
||
|
|
3, # float (numeric value between 0.1 and 20) in 'Guidance Scale' Slider component
|
||
|
|
True, # bool in 'Randomize seed' Checkbox component
|
||
|
|
api_name="/run")
|
||
|
|
|
||
|
|
# Путь к изображению
|
||
|
|
return result[0][0]["image"]
|