Update app.py

app.py

@@ -129,6 +129,8 @@ def predict(Weight_in_Kgs, Heart_rate, pulse_rate, Systolic_BP, Diastolic_BP, BI
     _, _, _, _, pred_dosage, pred_medicine_class, pred_pain_cls = model(X)
     # Get the medicine name from the model output
     medicine, dosage_level_nm = get_medicine_name(pred_medicine_class, pred_pain_cls)
+    if Pain_Position =='No Pain':
+        medicine, dosage_level_nm = 'NA', 'NA'
     # Return the predicted medicine name as a string
     return "The predicted medicine for you is: " + medicine[0], dosage_level_nm[0]