create requirements.txt

requirements.txt

@@ -0,0 +1,28 @@
+# Core dependencies for AI4Bharat IndicTrans2 project
+torch>=2.0.0
+transformers>=4.30.0
+IndicTransToolkit>=1.0.0
+
+# Optional GPU acceleration (recommended for better performance)
+# Note: flash-attn requires CUDA and should be installed separately if needed:
+# pip install flash-attn>=2.0.0
+# flash-attn>=2.0.0
+
+# Development and testing dependencies
+pytest>=7.0.0
+pytest-cov>=4.0.0
+
+# Utility packages
+numpy>=1.21.0
+huggingface-hub>=0.34.0,<1.0
+
+# For handling different data formats (if needed)
+datasets>=2.10.0
+sentencepiece>=0.1.97
+
+# For logging and monitoring (optional)
+wandb>=0.15.0
+tensorboard>=2.10.0
+
+#For UI and demo purposes (optional)
+gradio>=4.0.0