rdezwart commited on
Commit
6879c60
·
1 Parent(s): e881756

Start 'selected_image' at 0

Browse files

Was getting "AttributeError: 'NoneType' object has no attribute '_id'", let's see if this fixes it.

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ yolos_id = "hustvl/yolos-small-300"
22
  yolos_processor: YolosImageProcessor = YolosImageProcessor.from_pretrained(yolos_id)
23
  yolos_model: YolosForObjectDetection = YolosForObjectDetection.from_pretrained(yolos_id)
24
 
25
- selected_image: int | None = None
26
 
27
 
28
  def answer_question(img, prompt):
 
22
  yolos_processor: YolosImageProcessor = YolosImageProcessor.from_pretrained(yolos_id)
23
  yolos_model: YolosForObjectDetection = YolosForObjectDetection.from_pretrained(yolos_id)
24
 
25
+ selected_image = 0
26
 
27
 
28
  def answer_question(img, prompt):