Spaces:
Sleeping
Sleeping
AdilzhanB
commited on
Commit
·
9c08496
1
Parent(s):
80af471
App change
Browse files
app.py
CHANGED
|
@@ -45,7 +45,7 @@ def predict_bone_age(image, gender):
|
|
| 45 |
img_tensor = transform(image).unsqueeze(0).numpy()
|
| 46 |
# Gender: 0=male, 1=female
|
| 47 |
gender_val = 0.0 if gender.lower() == "male" else 1.0
|
| 48 |
-
gender_tensor = np.array([
|
| 49 |
|
| 50 |
# ONNX inference
|
| 51 |
outputs = ort_session.run(None, {"image": img_tensor, "gender": gender_tensor})
|
|
|
|
| 45 |
img_tensor = transform(image).unsqueeze(0).numpy()
|
| 46 |
# Gender: 0=male, 1=female
|
| 47 |
gender_val = 0.0 if gender.lower() == "male" else 1.0
|
| 48 |
+
gender_tensor = np.array([gender_val], dtype=np.float32)
|
| 49 |
|
| 50 |
# ONNX inference
|
| 51 |
outputs = ort_session.run(None, {"image": img_tensor, "gender": gender_tensor})
|