vikhyatk commited on
Commit
7219c6d
·
verified ·
1 Parent(s): 4e26b97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -164,7 +164,7 @@ def detect(img, object):
164
  img = Resize(768)(img)
165
  w, h = img.size
166
 
167
- objs = moondream.detect(img, object)
168
  draw_image = ImageDraw.Draw(img)
169
  for o in objs:
170
  draw_image.rectangle(
@@ -189,7 +189,7 @@ def point(img, object):
189
  img = Resize(768)(img)
190
  w, h = img.size
191
 
192
- objs = moondream.point(img, object, tokenizer)
193
  draw_image = ImageDraw.Draw(img)
194
  for o in objs:
195
  draw_image.ellipse(
 
164
  img = Resize(768)(img)
165
  w, h = img.size
166
 
167
+ objs = moondream.detect(img, object)["objects"]
168
  draw_image = ImageDraw.Draw(img)
169
  for o in objs:
170
  draw_image.rectangle(
 
189
  img = Resize(768)(img)
190
  w, h = img.size
191
 
192
+ objs = moondream.point(img, object)["points"]
193
  draw_image = ImageDraw.Draw(img)
194
  for o in objs:
195
  draw_image.ellipse(