saliacoel commited on
Commit
1b68d46
·
verified ·
1 Parent(s): 6faf665

Upload Salia_Bam_2.py

Browse files
Files changed (1) hide show
  1. Salia_Bam_2.py +8 -3
Salia_Bam_2.py CHANGED
@@ -69,7 +69,8 @@ class Salia_BAM_2:
69
 
70
  Outputs:
71
  Frontview_pos, Sideview_pos, Diagonal_pos, Rearview_pos, image,
72
- Frontview_neg, Sideview_neg, Diagonal_neg, Rearview_neg, BAM_out
 
73
 
74
  BAM_in:
75
  - If BAM_in == "" (exactly empty), behave normally (use UI inputs).
@@ -698,12 +699,14 @@ class Salia_BAM_2:
698
  RETURN_TYPES = (
699
  "STRING", "STRING", "STRING", "STRING", "IMAGE",
700
  "STRING", "STRING", "STRING", "STRING",
701
- "STRING" # NEW: BAM_out
 
702
  )
703
  RETURN_NAMES = (
704
  "Frontview_pos", "Sideview_pos", "Diagonal_pos", "Rearview_pos", "image",
705
  "Frontview_neg", "Sideview_neg", "Diagonal_neg", "Rearview_neg",
706
- "BAM_out"
 
707
  )
708
  FUNCTION = "run"
709
 
@@ -825,6 +828,7 @@ class Salia_BAM_2:
825
  # ---------- Bucket selection from age+gender ----------
826
  gender_norm = _parse_gender(gender_in)
827
  n = _parse_number(age_in)
 
828
  bucket = _bucket_from_number(n)
829
  fname = f"{gender_norm}{bucket}.png"
830
 
@@ -1091,6 +1095,7 @@ class Salia_BAM_2:
1091
  Diagonal_neg,
1092
  Rearview_neg,
1093
  BAM_out,
 
1094
  )
1095
 
1096
 
 
69
 
70
  Outputs:
71
  Frontview_pos, Sideview_pos, Diagonal_pos, Rearview_pos, image,
72
+ Frontview_neg, Sideview_neg, Diagonal_neg, Rearview_neg, BAM_out,
73
+ age (INT)
74
 
75
  BAM_in:
76
  - If BAM_in == "" (exactly empty), behave normally (use UI inputs).
 
699
  RETURN_TYPES = (
700
  "STRING", "STRING", "STRING", "STRING", "IMAGE",
701
  "STRING", "STRING", "STRING", "STRING",
702
+ "STRING", # BAM_out
703
+ "INT", # age
704
  )
705
  RETURN_NAMES = (
706
  "Frontview_pos", "Sideview_pos", "Diagonal_pos", "Rearview_pos", "image",
707
  "Frontview_neg", "Sideview_neg", "Diagonal_neg", "Rearview_neg",
708
+ "BAM_out",
709
+ "age",
710
  )
711
  FUNCTION = "run"
712
 
 
828
  # ---------- Bucket selection from age+gender ----------
829
  gender_norm = _parse_gender(gender_in)
830
  n = _parse_number(age_in)
831
+ age_out_int = 0 if n is None else int(n)
832
  bucket = _bucket_from_number(n)
833
  fname = f"{gender_norm}{bucket}.png"
834
 
 
1095
  Diagonal_neg,
1096
  Rearview_neg,
1097
  BAM_out,
1098
+ age_out_int,
1099
  )
1100
 
1101