Moved device assignement to beginning of function

scripts.py

@@ -118,11 +118,12 @@ def prepare_training_datasets():
     consistency.to_csv('corpora/consistency.csv')
     
 def train_models(path,progress=gradio.Progress(track_tqdm=True)):
+    device = torch.device('cuda:0')
     torch.cuda.empty_cache()
     tokenizer = tokenizers.Tokenizer.from_pretrained('roberta-base')
     trainer = qarac.models.QaracTrainerModel.QaracTrainerModel('roberta-base', 
                                                                tokenizer)
-    device = torch.device('cuda:0')
+    
     trainer.to(device)
     loss_fn = CombinedLoss()
     loss_fn.cuda()