Spaces:
Sleeping
Sleeping
Create requirements.txt
Browse files- 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
|