Update requirements.txt
Browse files- requirements.txt +5 -4
requirements.txt
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
-
gradio==
|
| 2 |
-
transformers==4.
|
| 3 |
-
|
| 4 |
-
|
|
|
|
|
|
| 1 |
+
gradio==3.27.0 # For the Gradio interface
|
| 2 |
+
transformers==4.30.0 # Hugging Face transformers for NLP tasks
|
| 3 |
+
torch==2.1.0 # PyTorch (ensure compatibility with your model and CUDA version)
|
| 4 |
+
numpy<2 # If NumPy 2.x causes issues, this ensures compatibility
|
| 5 |
+
huggingface_hub==0.14.0 # For Hugging Face model downloading
|