Modify string output

app.py

@@ -71,7 +71,22 @@ def predict_bone_age(Radiograph, Sex):
         img = torch.cat([img, torch.zeros_like(img) - 1], dim=1)
     with torch.no_grad():
         bone_age = ensemble(img.float())[0].item()
-    return f"Estimated Bone Age: {int(bone_age)} years, {int(bone_age % int(bone_age) * 12)} months"
+    total_months = bone_age * 12
+    years = total_months // 12
+    months = total_months - years * 12
+    months = round(months)
+    if months == 12: 
+        years += 1
+        months = 0
+    if years == 0:
+        str_output = f"{months} months" if months != 1 else "1 month"
+    else:
+        months = round(months)
+        if months == 0:
+            str_output = f"{years} years" if years != 1 else "1 year"
+        else:
+            str_output = f"{years} years, {months} months" if months != 1 else f"{years} years, 1 month"
+    return f"Estimated Bone Age: {str_output}"
 
 
 image = gr.Image(shape=(512, 512), image_mode="L")