Upload threshold.py

python_Code/threshold.py

@@ -0,0 +1,16 @@
+import spacy
+
+def set_threshold(model_path, threshold):
+    # Load the trained model
+    nlp = spacy.load(model_path)
+
+    # Set the threshold for text classification
+    nlp.get_pipe('textcat_multilabel').threshold = threshold
+
+    return nlp
+
+# Example usage:
+if __name__ == "__main__":
+    model_path = "./my_trained_model"
+    threshold = 0.21
+    nlp = set_threshold(model_path, threshold)