fixed device detect issue

app.py

@@ -11,13 +11,13 @@ if __name__ == '__main__':
     import gradio as gr
     import os
 
-    device = f'cuda:{torch.cuda.current_device()}' if torch.cuda.is_available() else 'cpu'  
+    device = f'cuda' 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')
     model = CustomModel(CFG, config_path=config_path, pretrained=False)
     state = torch.load(model_path,
-                        map_location=torch.device('cuda'))
+                        map_location=device)
     model.load_state_dict(state['model'])
 
     def get_answer(context, feature):