multimodalart HF staff commited on
Commit
d837061
1 Parent(s): 7b93470

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -167,16 +167,16 @@ def infer(text):
167
 
168
  # idetnical to `infer` function without gradio state updates for share btn
169
  def infer_examples(text):
170
- with autocast("cuda"):
171
- images_list = pipe(
172
  [text]*2,
173
  num_inference_steps=50,
174
  guidance_scale=7.5
175
  )
176
- output_images = []
177
- for i, image in enumerate(images_list["sample"]):
178
- output_images.append(image)
179
- return output_images
180
 
181
  css = '''
182
  .gradio-container {font-family: 'IBM Plex Sans', sans-serif}
 
167
 
168
  # idetnical to `infer` function without gradio state updates for share btn
169
  def infer_examples(text):
170
+ #with autocast("cuda"):
171
+ images_list = pipe(
172
  [text]*2,
173
  num_inference_steps=50,
174
  guidance_scale=7.5
175
  )
176
+ #output_images = []
177
+ #for i, image in enumerate(images_list["sample"]):
178
+ # output_images.append(image)
179
+ return images_list.images
180
 
181
  css = '''
182
  .gradio-container {font-family: 'IBM Plex Sans', sans-serif}