Update app.py

app.py

@@ -5,7 +5,7 @@ import importlib.util
 import gradio as gr
 from rich import _console
 from transformers import AutoModel, AutoTokenizer, AutoConfig, pipeline
-
+import torch 
 from huggingface_hub import snapshot_download
 import sys, pathlib
 import os
@@ -13,6 +13,17 @@ os.environ["OPENAI_API_KEY"] = "test"
 os.environ["OMP_NUM_THREADS"] = "4"
 
 print("All imports finished")
+print(f"Python version: {sys.version}")
+print(f"PyTorch version: {torch.__version__}")
+print(f"CUDA available: {torch.cuda.is_available()}")
+print(f"CUDA version: {torch.version.cuda}")
+print(f"cuDNN version: {torch.backends.cudnn.version()}")
+print(f"Number of GPUs: {torch.cuda.device_count()}")
+
+if torch.cuda.is_available():
+    for i in range(torch.cuda.device_count()):
+        print(f"GPU {i}: {torch.cuda.get_device_name(i)}")
+        print(f"  Memory: {torch.cuda.get_device_properties(i).total_memory / 1e9:.2f} GB")
 
 # # 1) Download the repo to a local cache dir
 # print("Downloading remote vine repo...")
@@ -89,6 +100,8 @@ def process_video(video_file, categorical_keywords, unary_keywords, binary_keywo
         import traceback
         traceback.print_exc()
         raise
+
+    
     categorical_keywords = [kw.strip() for kw in categorical_keywords.split(",")] if categorical_keywords else []
     unary_keywords = [kw.strip() for kw in unary_keywords.split(",")] if unary_keywords else []
     binary_keywords = [kw.strip() for kw in binary_keywords.split(",")] if binary_keywords else []