Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
|
|
34 |
|
35 |
# Predict fracture
|
36 |
prediction = predict_fracture(xray)
|
37 |
-
diagnosed_class = "
|
38 |
|
39 |
# Injury severity classification
|
40 |
severity = "Mild" if prediction < 0.3 else "Moderate" if prediction < 0.7 else "Severe"
|
|
|
34 |
|
35 |
# Predict fracture
|
36 |
prediction = predict_fracture(xray)
|
37 |
+
diagnosed_class = "normal" if prediction > 0.5 else "Fractured"
|
38 |
|
39 |
# Injury severity classification
|
40 |
severity = "Mild" if prediction < 0.3 else "Moderate" if prediction < 0.7 else "Severe"
|