Create requirements.txt

requirements.txt

@@ -0,0 +1,31 @@
+gradio
+torch
+torchvision
+torchaudio
+transformers
+Pillow
+pandas
+matplotlib
+torchmetrics[image,multimodal]
+open_clip_torch
+accelerate # Может потребоваться для некоторых моделей
+timm # Может потребоваться для некоторых моделей
+
+# Зависимости для ONNX и deepghs моделей
+onnxruntime # или onnxruntime-gpu, если GPU доступен и настроен в Space
+huggingface_hub>=0.14.0
+# dghs-imgutils и hbutils могут быть не нужны напрямую, если мы реализуем их функции
+# но если функции rgb_encode сложны, лучше их иметь
+# Для простоты, сначала попробуем реализовать ключевые части без них, если возможно,
+# или добавим, если это необходимо для точного воспроизведения.
+# imgutils используется в предоставленном коде для rgb_encode
+# pip install imgutils --no-deps # Если dghs-imgutils не на PyPI, а imgutils - это оно
+# Попробуем найти imgutils или реализовать аналог rgb_encode
+numpy # Уже есть, но важен
+scikit-image # Может использоваться imgutils
+opencv-python>=4.6.0 # Может использоваться imgutils
+# natsort # Для сортировки моделей, если бы мы динамически их загружали
+
+# Уточнение: dghs-imgutils доступен на PyPI как 'imgutils' (скорее всего)
+# pip install imgutils
+imgutils