Upload server.py with huggingface_hub

server.py

@@ -16,7 +16,7 @@ print(f"[video] torch version: {torch.__version__}", flush=True)
 print(f"[video] CUDA available: {torch.cuda.is_available()}", flush=True)
 if torch.cuda.is_available():
     print(f"[video] GPU: {torch.cuda.get_device_name(0)}", flush=True)
-    print(f"[video] VRAM: {torch.cuda.get_device_properties(0).total_mem / 1e9:.1f} GB", flush=True)
+    print(f"[video] VRAM: {torch.cuda.get_device_properties(0).total_memory / 1e9:.1f} GB", flush=True)
 
 app = Flask(__name__)
 pipe = None