File size: 1,259 Bytes
d154a5b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
c3cddd1
d154a5b
c3cddd1
d154a5b
 
c3cddd1
e1e5e39
 
 
 
 
 
c8776ee
e1e5e39
 
 
 
d154a5b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
# 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)"}
  ]
}