Update app.py

app.py

@@ -64,7 +64,8 @@ def save_ply_from_array(verts):
     tmpf.flush(); tmpf.close()
     return tmpf.name
 
-@spaces.GPU(duration=120)
+#@spaces.GPU(duration=120)
+@spaces.GPU
 def predict(_chatbot,task_history,viewer_voxel,viewer_mesh,task_new,seed,top_k,top_p,temperature):
     torch.manual_seed(seed)
     chat_query = _chatbot[-1][0]
@@ -393,6 +394,10 @@ def _transform_messages(original_messages):
 
     return transformed_messages
 
+print(f"CUDA Available: {torch.cuda.is_available()}")
+print(f"CUDA Version: {torch.version.cuda}")
+print(f"Number of GPUs: {torch.cuda.device_count()}")
+    
 from trellis.models.sparse_structure_vqvae import VQVAE3D
 device       = torch.device("cuda")
 vqvae        = VQVAE3D(num_embeddings=8192)