Update api_schemas.py

api_schemas.py

@@ -1,17 +1,26 @@
-# api_schemas.py
-DESCRIBE_RAW_MULTIPART = {
-  "method": "POST",
-  "path": "/api/describe_raw",
-  "multipart_fields": [
-    {"name": "image", "type": "file"},
-    {"name": "text", "type": "text", "default": "Describe la imagen con detalle."},
-    {"name": "max_new_tokens", "type": "int", "default": 256},
-    {"name": "temperature", "type": "float", "default": 0.7}
-  ]
-}
-
-PREDICT_JSON = {
-  "method": "POST",
-  "path": "/api/predict/describe",
-  "json_body": {"data": ["<file or url>", "prompt text", 256, 0.7]}
-}
+# api_schemas.py
+DESCRIBE_RAW_MULTIPART = {
+  "method": "POST",
+  "path": "/api/describe_raw",
+  "multipart_fields": [
+    {"name": "image", "type": "file"},
+    {"name": "text", "type": "text", "default": "Describe la imagen con detalle."},
+    {"name": "max_new_tokens", "type": "int", "default": 256},
+    {"name": "temperature", "type": "float", "default": 0.7}
+  ]
+}
+
+PREDICT_JSON = {
+  "method": "POST",
+  "path": "/api/predict/describe",
+  "json_body": {"data": ["<file or url>", "prompt text", 256, 0.7]}
+}
+
+IMAGE_SIZE_MULTIPART = {
+  "method": "POST",
+  "path": "/api/image_size",
+  "multipart_fields": [
+    {"name": "image", "type": "file"}
+  ],
+  "response": {"data": ["<ancho>x<alto>"]}
+}