komleva commited on
Commit
af1adb2
·
1 Parent(s): be0ba6f

Update app

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -16,7 +16,7 @@ from utils.zero_shot_utils import zero_shot_step
16
  from tasks.mm_tasks.vqa_gen import VqaGenTask
17
  from tasks.mm_tasks.refcoco import RefcocoTask
18
  from models.ofa import OFAModel
19
- from PIL import Image,ImageDraw
20
  from torchvision import transforms
21
  import gradio as gr
22
 
@@ -168,8 +168,8 @@ def open_domain_vqa(Image, Question):
168
  if left == right:
169
  left = left -10
170
 
171
- draw = ImageDraw.Draw(Image)
172
- return draw.rectangle([left, top, right, bottom], outline="red")#Image.crop((left, top, right, bottom))#, result[0]['answer']
173
 
174
 
175
  title = "Visual Question Answering"
 
16
  from tasks.mm_tasks.vqa_gen import VqaGenTask
17
  from tasks.mm_tasks.refcoco import RefcocoTask
18
  from models.ofa import OFAModel
19
+ from PIL import Image
20
  from torchvision import transforms
21
  import gradio as gr
22
 
 
168
  if left == right:
169
  left = left -10
170
 
171
+
172
+ return Image.crop((left, top, right, bottom))#, result[0]['answer']
173
 
174
 
175
  title = "Visual Question Answering"