ciditel commited on
Commit
0ddb5fd
·
verified ·
1 Parent(s): 28fdde2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def run_example(task_prompt, image, text_input=None, model_id='microsoft/Florenc
44
  else:
45
  prompt = task_prompt + text_input
46
  iimage=image
47
- image=Image.open(image)
48
  inputs = processor(text=prompt, images=image, return_tensors="pt").to("cuda")
49
  generated_ids = model.generate(
50
  input_ids=inputs["input_ids"],
 
44
  else:
45
  prompt = task_prompt + text_input
46
  iimage=image
47
+ image=fig_to_pil(image)
48
  inputs = processor(text=prompt, images=image, return_tensors="pt").to("cuda")
49
  generated_ids = model.generate(
50
  input_ids=inputs["input_ids"],