QHL067 commited on
Commit
55482bc
·
1 Parent(s): 0236b52

save image

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -235,7 +235,7 @@ def infer(
235
 
236
  to_pil = ToPILImage()
237
 
238
- pil_image = to_pil(sample[0])
239
 
240
  return pil_image, seed
241
 
 
235
 
236
  to_pil = ToPILImage()
237
 
238
+ pil_image = to_pil(samples[0])
239
 
240
  return pil_image, seed
241