Create requirements.txt

requirements.txt

@@ -1,9 +1,6 @@
-omegaconf==2.3.0
-Pillow==10.1.0
-einops==0.7.0
-torch==2.0.1
-transformers==4.35.0
-trimesh==4.0.5
-rembg
-huggingface-hub
-gradio
+--index-url https://download.pytorch.org/whl/cu118
+torch
+torchvision
+torchaudio
+transformers
+