Update requirements.txt

requirements.txt

@@ -1,8 +1,24 @@
-   setfit>=1.0.0
-   datasets>=2.14.0
-   sentence-transformers>=2.2.0
-   transformers>=4.30.0
-   pandas>=2.0.0
-   numpy>=1.24.0
-   gradio>=4.0.0
-   huggingface_hub>=0.16.0
+# Gradio Space Requirements for Hugging Face Spaces
+# This file is used when deploying to Hugging Face Spaces
+
+# Core SetFit library
+setfit>=1.0.0
+
+# Hugging Face ecosystem
+datasets>=2.14.0
+sentence-transformers>=2.2.0
+transformers>=4.30.0
+
+# Data handling
+pandas>=2.0.0
+numpy>=1.24.0
+
+# Gradio for the web interface
+gradio>=4.0.0
+
+# Hugging Face Hub (for loading models)
+huggingface_hub>=0.16.0
+
+# Scikit-learn (must match training version for model compatibility)
+# Model was trained with scikit-learn 1.8.0, need >=1.8.0 for predict_proba to work
+scikit-learn>=1.8.0