Rooni commited on
Commit
f3e26fa
·
1 Parent(s): fae13e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import gradio as gr
2
  import requests
3
  import os
 
 
4
 
5
  # API ссылка
6
  url = os.getenv("URL")
@@ -24,6 +26,7 @@ def render(prompt, negative_prompt, output_format, output_size, style, lighting,
24
  response = requests.post(url, headers=headers, json=data)
25
  if response.status_code == 200:
26
  return response.content
 
27
  else:
28
  return None
29
 
 
1
  import gradio as gr
2
  import requests
3
  import os
4
+ import io
5
+ from PIL import Image
6
 
7
  # API ссылка
8
  url = os.getenv("URL")
 
26
  response = requests.post(url, headers=headers, json=data)
27
  if response.status_code == 200:
28
  return response.content
29
+ image = Image.open(io.BytesIO(response.content))
30
  else:
31
  return None
32