svision / api_schemas.py
VeuReu's picture
Update api_schemas.py
c8776ee verified
raw
history blame
1.26 kB
# 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]}
}
FACE_IMAGE_EMBEDDING = {
"method": "POST",
"path": "/api/face_image_embedding",
"multipart_fields": [
{"name": "image", "type": "file"}
]
}
SCENES_EXTRACTION = {
"method": "POST",
"path": "/api/scenes_extraction",
"multipart_fields": [
{"name": "video_file", "type": "str"},
{"name": "threshold", "type": "float", "default": 30.0, "description": "Sensibilidad para identificar escenas diferentes"},
{"name": "offset_frames", "type": "int", "default": 10, "description": "En vez de te coja la primera imagen de la escena te la coja un poco avanzada en la escena" },
{"name": "crop_ratio", "type": "float", "default": 0.1, "description": "Recorte en los bordes de la imagen (para evitar marcas de agua)"}
]
}