Update handler.py

Added age-individ parameter

handler.py

@@ -47,8 +47,17 @@ class EndpointHandler:
             elif k=='down':
                 if v:
                     prompt+=f'{multi_modal_prefix}<{k}>{v}</{k}>' if isinstance(v, str) else f'{multi_modal_prefix}<{k}>{" ".join(v)} </{k}>'
+            elif k=='age':
+                if isinstance(v, int):
+                    if prompt_config['species'].strip() == 'human':
+                        prompt+=f'<{k}_individ>{v} </{k}_individ>'
+                    elif prompt_config['species'].strip() == 'macaque':
+                        prompt+=f'<{k}_individ>Macaca-{int(v/20)} </{k}_individ>'
             else:
-                prompt+=f'<{k}>{v}</{k}>' if isinstance(v, str) else f'<{k}>{" ".join(v)} </{k}>'
+                if v:
+                    prompt+=f'<{k}>{v.strip()} </{k}>' if isinstance(v, str) else f'<{k}>{" ".join(v)} </{k}>'
+                else:
+                    prompt+=f'<{k}></{k}>'
         return prompt
 
     def custom_generate(self,