Ashoka74 commited on
Commit
52a473a
·
verified ·
1 Parent(s): f6f6a12

Update app_2.py

Browse files
Files changed (1) hide show
  1. app_2.py +2 -1
app_2.py CHANGED
@@ -1232,11 +1232,12 @@ with block:
1232
  example_quick_prompts.click(lambda x, y: ', '.join(y.split(', ')[:2] + [x[0]]), inputs=[example_quick_prompts, prompt], outputs=prompt, show_progress=False, queue=False)
1233
  example_quick_subjects.click(lambda x: x[0], inputs=example_quick_subjects, outputs=prompt, show_progress=False, queue=False)
1234
 
 
1235
  run_button.click(
1236
  fn=infer,
1237
  inputs=[
1238
  text_prompt,
1239
- extracted_fg,
1240
  ],
1241
  outputs=[orientation_result],
1242
  )
 
1232
  example_quick_prompts.click(lambda x, y: ', '.join(y.split(', ')[:2] + [x[0]]), inputs=[example_quick_prompts, prompt], outputs=prompt, show_progress=False, queue=False)
1233
  example_quick_subjects.click(lambda x: x[0], inputs=example_quick_subjects, outputs=prompt, show_progress=False, queue=False)
1234
 
1235
+ angles_fg = Image.fromarray(extracted_fg, 'RGB')
1236
  run_button.click(
1237
  fn=infer,
1238
  inputs=[
1239
  text_prompt,
1240
+ angles_fg,
1241
  ],
1242
  outputs=[orientation_result],
1243
  )