Upload inference.py

inference.py

@@ -92,7 +92,7 @@ def convert_intent_to_label(intent_logit):
     return intent_labels[int(intent_logit)]
 
 
-def main(input_data):
+def main():
     """
     Main function to perform inference using the pre-trained model.
     """
@@ -100,6 +100,7 @@ def main(input_data):
     model_path = "pytorch_model.bin"
     model = load_model(model_path)
 
+    input_data = input("Enter the text to analyze: ")
     # Preprocess the input text
     input_ids, attention_mask, offset_mapping = preprocess_input(input_data)
 
@@ -113,13 +114,9 @@ def main(input_data):
     ner_logits = align_ner_predictions_with_input(ner_logits, offset_mapping, input_data)
     intent_label = convert_intent_to_label(intent_logits)
 
-    return ner_logits, intent_label
+    print(f"Ner logits: {ner_logits}")
+    print(f"Intent logits: {intent_label}")
     
 
 if __name__ == "__main__":
-
-    input_data = "I want to schedule a meeting for the 15th of this month at 2:30 PM."
-    ner_logits, intent_label = main(input_data)
-
-    print(f"Ner logits: {ner_logits}")
-    print(f"Intent logits: {intent_label}")
+    main()