Update model.py

model.py

@@ -234,6 +234,12 @@ def get_predictions(feature_text, pn_history):
     pn_history = pn_history.lower()
     
     location_preds, pred_string = predict_location_preds(tokenizer, model, [feature_text], [pn_history])
+    
+    if pred_string == "":
+        pred_string = 'Feature not present!'
+    else:
+        pred_string = 'Feature is present!' + '\nText Span - ' + pred_string
+    
     return pred_string
 
 tokenizer = AutoTokenizer.from_pretrained(config['tokenizer_path'])
@@ -246,9 +252,4 @@ model.load_state_dict(
         map_location=torch.device(config['device'])
     )
 )
-model.eval()
-
-# input_text = create_sample_test()
-# feature_text = input_text.feature_text[0]
-# pn_history = input_text.pn_history[0]
-# get_predictions(feature_text, pn_history)
+model.eval()