Update app.py

app.py

@@ -58,7 +58,7 @@ class DosagePredictionModel(nn.Module):
         pred_pain_cls_y = torch.argmax(pain_output_cls, dim = 1)
         
         if (classification_targets is None) or (regression_targets is None) or (pain_cls_tgt is None):
-            print('Inference Mode')
+            #print('Inference Mode')
             loss, regression_loss, med_cls_loss, pain_cls_loss = None, None, None, None
         else:
             loss_fn = nn.SmoothL1Loss()  # mean square error