Rooni commited on
Commit
ee43da4
·
verified ·
1 Parent(s): d46b53f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -3,8 +3,6 @@ import gradio as gr
3
  import numpy as np
4
  from gradio_client import Client
5
  from PIL import Image
6
- import requests
7
- from io import BytesIO
8
 
9
  # Используем переменные окружения для гибкости
10
  MODEL_ID = os.getenv("MODEL_ID", "KingNish/SDXL-Flash")
@@ -140,9 +138,7 @@ with gr.Blocks(css=css) as demo:
140
  use_resolution_binning=True,
141
  api_name="/run"
142
  )
143
- image_url = response[0] # Предполагаем, что в ответе содержится URL изображения
144
- image = Image.open(BytesIO(requests.get(image_url).content)) # Загружаем изображение из URL
145
- results.append(image)
146
  return results, seed
147
 
148
  gr.on(
 
3
  import numpy as np
4
  from gradio_client import Client
5
  from PIL import Image
 
 
6
 
7
  # Используем переменные окружения для гибкости
8
  MODEL_ID = os.getenv("MODEL_ID", "KingNish/SDXL-Flash")
 
138
  use_resolution_binning=True,
139
  api_name="/run"
140
  )
141
+ results.append((response[0], prompt)) # Возвращаем изображение и подпись
 
 
142
  return results, seed
143
 
144
  gr.on(