Utsaha commited on
Commit
55d6996
1 Parent(s): 0ab3c34

Create requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -0
requirements.txt ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ torch==2.0.1 # Ensure that you have a compatible version for CUDA if using GPU
2
+ torchaudio==2.0.1 # Ensure compatibility with the PyTorch version
3
+ transformers==4.29.2 # Version of the Hugging Face Transformers library
4
+ scikit-learn==1.2.2 # For LabelEncoder
5
+ tf-keras==2.11.0 # TensorFlow Keras, version may vary based on your TensorFlow installation
6
+ pandas==2.0.1 # For handling dataframes