Update app.py

app.py

@@ -14,6 +14,12 @@
 
 import gradio as gr
 import spaces
+
+import torch
+import types
+torch.cuda.get_device_capability = lambda *args, **kwargs: (8, 6)
+torch.cuda.get_device_properties = lambda *args, **kwargs: types.SimpleNamespace(name='NVIDIA A10G', major=8, minor=6, total_memory=23836033024, multi_processor_count=80)
+
 import argparse
 from model import SALMONN