Update requirements.txt
Browse files- requirements.txt +12 -12
requirements.txt
CHANGED
@@ -1,12 +1,12 @@
|
|
1 |
-
torch
|
2 |
-
torchvision
|
3 |
-
ultralytics # For YOLOv8 models
|
4 |
-
transformers # For Hugging Face model interfacing
|
5 |
-
huggingface_hub # For Hugging Face model uploading and management
|
6 |
-
pandas # For data manipulation
|
7 |
-
numpy # For numerical operations
|
8 |
-
opencv-python-headless # For YOLOv8 image processing
|
9 |
-
scipy # For scientific computations
|
10 |
-
requests # For handling HTTP requests
|
11 |
-
gradio
|
12 |
-
Pillow
|
|
|
1 |
+
torch # Ensure compatibility with your GPU and cu118
|
2 |
+
torchvision # Ensure compatibility with the specified torch version
|
3 |
+
ultralytics # For YOLOv8 models
|
4 |
+
transformers # For Hugging Face model interfacing
|
5 |
+
huggingface_hub # For Hugging Face model uploading and management
|
6 |
+
pandas # For data manipulation
|
7 |
+
numpy # For numerical operations
|
8 |
+
opencv-python-headless # For YOLOv8 image processing
|
9 |
+
scipy # For scientific computations
|
10 |
+
requests # For handling HTTP requests
|
11 |
+
gradio
|
12 |
+
Pillow
|