Update app.py

app.py

@@ -93,8 +93,12 @@ def predict_risk(age, sex, cd4_count, viral_load, wbc_count, hemoglobin, platele
         # 4. Process Output
         probabilities = ort_outs[0][0]
         risk_probability = probabilities[1] # Probability of High Risk
-
-        return f"HIV Risk Probability: {risk_probability:.4f}"
+        if 0 < risk_probability <= 100:
+            return f"HIV Risk Probability: {risk_probability:.4f}"
+        elif risk_probability > 100:
+            return f"HIV Risk Probability: 100"
+        else:
+            return f"HIV Risk Probability: 0"
 
     except Exception as e:
         logging.exception(f"An error occurred during prediction: {e}")