bug fix

app.py

@@ -11,7 +11,7 @@ if __name__ == '__main__':
     import gradio as gr
     import os
 
-    device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'  
+    device = f'cuda:{torch.cuda.current_device()}' if torch.cuda.is_available() else 'cpu'  
     config_path = os.path.join('models_file', 'config.pth')
     model_path  = os.path.join('models_file', 'microsoft-deberta-base_0.9449373420387531_8_best.pth')
     tokenizer = AutoTokenizer.from_pretrained('models_file/tokenizer')