root-sajjan
commited on
Commit
•
9af57d5
1
Parent(s):
4f45aec
sending image urls too
Browse files
model.py
CHANGED
@@ -159,6 +159,7 @@ class YOLOModel:
|
|
159 |
"price":result_llms["price"],
|
160 |
"details":result_llms["description"],
|
161 |
"detected_text":detected_text,
|
|
|
162 |
})
|
163 |
|
164 |
valid_indices = set(range(len(predictions)))
|
|
|
159 |
"price":result_llms["price"],
|
160 |
"details":result_llms["description"],
|
161 |
"detected_text":detected_text,
|
162 |
+
"image_url":image_url
|
163 |
})
|
164 |
|
165 |
valid_indices = set(range(len(predictions)))
|