Spaces:
Running
on
T4
Running
on
T4
Update aya_vision_utils.py
Browse files- aya_vision_utils.py +6 -4
aya_vision_utils.py
CHANGED
|
@@ -54,13 +54,15 @@ def get_aya_vision_response(incoming_message, image_filepath, max_size_mb=5):
|
|
| 54 |
print("image_filepath:", image_filepath)
|
| 55 |
max_size_bytes = max_size_mb * 1024 * 1024
|
| 56 |
|
| 57 |
-
|
|
|
|
|
|
|
| 58 |
image_type="image/jpeg"
|
| 59 |
-
elif
|
| 60 |
image_type = "image/png"
|
| 61 |
-
elif
|
| 62 |
image_type="image/webp"
|
| 63 |
-
elif
|
| 64 |
image_type="image/gif"
|
| 65 |
|
| 66 |
response=""
|
|
|
|
| 54 |
print("image_filepath:", image_filepath)
|
| 55 |
max_size_bytes = max_size_mb * 1024 * 1024
|
| 56 |
|
| 57 |
+
image_ext = image_filepath.lower()
|
| 58 |
+
|
| 59 |
+
if image_ext.endswith(".jpg") or image_ext.endswith('.jpeg'):
|
| 60 |
image_type="image/jpeg"
|
| 61 |
+
elif image_ext.endswith(".png"):
|
| 62 |
image_type = "image/png"
|
| 63 |
+
elif image_ext.endswith(".webp"):
|
| 64 |
image_type="image/webp"
|
| 65 |
+
elif image_ext.endswith(".gif"):
|
| 66 |
image_type="image/gif"
|
| 67 |
|
| 68 |
response=""
|