MatteoScript commited on
Commit
e070c28
·
verified ·
1 Parent(s): b30cb0a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -4
main.py CHANGED
@@ -211,10 +211,10 @@ def generate_image(request: Request, input_data: InputImage):
211
  # api_name="/run"
212
  #)
213
  result = client.predict(
214
- input_data.input # str in 'Prompt' Textbox component
215
  input_data.negativePrompt, # str in 'Negative prompt' Textbox component
216
  "(No style)", # str in 'Image Style' Radio component
217
- "" # str in 'Prompt 2' Textbox component
218
  "", # str in 'Negative prompt 2' Textbox component
219
  True, # bool in 'Use negative prompt' Checkbox component
220
  False, # bool in 'Use prompt 2' Checkbox component
@@ -229,8 +229,7 @@ def generate_image(request: Request, input_data: InputImage):
229
  True, # bool in 'Apply refiner' Checkbox component
230
  True, # bool in 'Randomize seed' Checkbox component
231
  api_name="/run"
232
- )
233
-
234
  image_url = result
235
  with open(image_url, 'rb') as img_file:
236
  img_binary = img_file.read()
 
211
  # api_name="/run"
212
  #)
213
  result = client.predict(
214
+ input_data.input, # str in 'Prompt' Textbox component
215
  input_data.negativePrompt, # str in 'Negative prompt' Textbox component
216
  "(No style)", # str in 'Image Style' Radio component
217
+ "", # str in 'Prompt 2' Textbox component
218
  "", # str in 'Negative prompt 2' Textbox component
219
  True, # bool in 'Use negative prompt' Checkbox component
220
  False, # bool in 'Use prompt 2' Checkbox component
 
229
  True, # bool in 'Apply refiner' Checkbox component
230
  True, # bool in 'Randomize seed' Checkbox component
231
  api_name="/run"
232
+ )
 
233
  image_url = result
234
  with open(image_url, 'rb') as img_file:
235
  img_binary = img_file.read()