Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 import subprocess
 import sys
 import time
-
+import torch
 
 import gradio as gr
 import requests
@@ -20,6 +20,11 @@ from llava.utils import (
     violates_moderation,
 )
 
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+# True
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+# Tesla T4
+
 logger = build_logger("gradio_web_server", "gradio_web_server.log")
 
 headers = {"User-Agent": "LLaVA Client"}