Rooni commited on
Commit
76109aa
·
1 Parent(s): 406fedb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -15
app.py CHANGED
@@ -5,22 +5,20 @@ import io
5
  from PIL import Image
6
 
7
  # API ссылка
8
- url = os.getenv("URL")
9
- key = os.getenv("KEY")
10
 
11
 
12
  # Функция для отправки запроса
13
- def render(prompt, negative_prompt, output_format, output_size, style, lighting, background, camera_position, camera_angle):
14
- headers = {"Authorization": f"Bearer {key}"}
15
  data = {
16
- "key": "your_api_key",
17
- "model_id": "realistic-vision-51",
18
  "prompt": prompt,
19
  "negative_prompt": negative_prompt,
20
- "width": "512",
21
- "height": "512",
22
  "samples": "1",
23
- "num_inference_steps": "30",
24
  "safety_checker": "no",
25
  "enhance_prompt": "yes",
26
  "seed": None,
@@ -48,12 +46,10 @@ gr.Interface(
48
  gr.Textbox(placeholder="Введите описание изображения"),
49
  gr.Textbox(placeholder="Введите отрицательный образ"),
50
  gr.Dropdown(["png", "jpg", "webp", "gif"]),
51
- gr.Dropdown(["256x256", "512x512", "1024x1024", "2048x2048"]),
52
- gr.Dropdown(["photorealistic", "cartoon", "anime", "sketch", "painting"]),
53
- gr.Dropdown(["natural", "studio", "night", "random"]),
54
- gr.Dropdown(["white", "black", "random"]),
55
- gr.Dropdown(["front", "top", "side"]),
56
- gr.Dropdown(["0", "45", "90"]),
57
  ],
58
  outputs=gr.Image()
59
  ).launch()
 
5
  from PIL import Image
6
 
7
  # API ссылка
8
+ url = "https://api-inference.huggingface.co/models/stablediffusionapi/realistic-vision-51"
 
9
 
10
 
11
  # Функция для отправки запроса
12
+ def render(prompt, negative_prompt, output_format, width, height, upscale, api_key):
 
13
  data = {
14
+ "key": api_key,
15
+ "model_id": "realistic-vision-51",
16
  "prompt": prompt,
17
  "negative_prompt": negative_prompt,
18
+ "width": width,
19
+ "height": height,
20
  "samples": "1",
21
+ "num_inference_steps": "40",
22
  "safety_checker": "no",
23
  "enhance_prompt": "yes",
24
  "seed": None,
 
46
  gr.Textbox(placeholder="Введите описание изображения"),
47
  gr.Textbox(placeholder="Введите отрицательный образ"),
48
  gr.Dropdown(["png", "jpg", "webp", "gif"]),
49
+ gr.Dropdown(["256", "512", "1024", "2048"]),
50
+ gr.Dropdown(["256", "512", "1024", "2048"]),
51
+ gr.Dropdown(["yes", "no"]),
52
+ gr.Textbox(placeholder="API_KEY"),
 
 
53
  ],
54
  outputs=gr.Image()
55
  ).launch()