Update app.py

app.py

@@ -1,11 +1,9 @@
+import gradio as gr
 import os
-os.environ["TOKENIZERS_PARALLELISM"] = "false"
 import spaces
 import subprocess
-import os
 import torch
 from threading import Thread
-import gradio as gr
 from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor,TextIteratorStreamer,AutoTokenizer
 from qwen_vl_utils import process_vision_info
 from trellis.pipelines import TrellisImageTo3DPipeline,TrellisTextTo3DPipeline
@@ -21,7 +19,12 @@ import random
 import open3d as o3d
 from huggingface_hub import hf_hub_download
 import numpy as np
+HF_TOKEN = os.environ.get("HF_TOKEN", None)
 
+import torch
+print("CUDA available:", torch.cuda.is_available())
+print("CUDA device count:", torch.cuda.device_count())
+1/0
 def _remove_image_special(text):
     text = text.replace('<ref>', '').replace('</ref>', '')
     return re.sub(r'<box>.*?(</box>|$)', '', text)