Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -903,6 +903,7 @@ def predict_middle(image, enable_fillet, fillet_value_mm):
|
|
903 |
def predict_full(image, enable_fillet, fillet_value_mm, enable_finger_cut, selected_outputs):
|
904 |
"""
|
905 |
Returns DXF + conditionally selected additional outputs
|
|
|
906 |
"""
|
907 |
radius = fillet_value_mm if enable_fillet == "On" else 0
|
908 |
finger_flag = "On" if enable_finger_cut == "On" else "Off"
|
@@ -916,17 +917,13 @@ def predict_full(image, enable_fillet, fillet_value_mm, enable_finger_cut, selec
|
|
916 |
finger_clearance=finger_flag,
|
917 |
)
|
918 |
|
919 |
-
#
|
920 |
-
|
921 |
-
|
922 |
-
|
923 |
-
|
924 |
-
|
925 |
-
|
926 |
-
if "Mask" in selected_outputs:
|
927 |
-
results.append(mask)
|
928 |
-
|
929 |
-
return results
|
930 |
|
931 |
|
932 |
|
|
|
903 |
def predict_full(image, enable_fillet, fillet_value_mm, enable_finger_cut, selected_outputs):
|
904 |
"""
|
905 |
Returns DXF + conditionally selected additional outputs
|
906 |
+
Always returns exactly 4 values to match output components
|
907 |
"""
|
908 |
radius = fillet_value_mm if enable_fillet == "On" else 0
|
909 |
finger_flag = "On" if enable_finger_cut == "On" else "Off"
|
|
|
917 |
finger_clearance=finger_flag,
|
918 |
)
|
919 |
|
920 |
+
# Always return 4 values to match the 4 output components
|
921 |
+
return (
|
922 |
+
dxf_path, # Always return DXF
|
923 |
+
ann if "Annotated Image" in selected_outputs else None,
|
924 |
+
outlines if "Outlines" in selected_outputs else None,
|
925 |
+
mask if "Mask" in selected_outputs else None
|
926 |
+
)
|
|
|
|
|
|
|
|
|
927 |
|
928 |
|
929 |
|