add

app.py

@@ -8,8 +8,7 @@ import spaces
 import sys
 import torch
 
-os.system(f"pip install \"git+https://github.com/facebookresearch/pytorch3d.git\"")
-
+os.system("pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt221/download.html")
 import torch
 import torch.nn as nn
 import gradio as gr

requirements.txt

@@ -1,5 +1,5 @@
-torch --index-url https://download.pytorch.org/whl/cu118
-torchvision --index-url https://download.pytorch.org/whl/cu118
+torch==2.2.1
+torchvision==0.17.1
 gradio>=4.13.0,<5.0.0
 diffusers
 transformers