Update src/translate/Translate.py

src/translate/Translate.py

@@ -7,6 +7,7 @@ import src.exception.Exception.Exception as ExceptionCustom
 from transformers import pipeline
 
 METHOD = "TRANSLATE"
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 def paraphraseTranslateMethod(requestValue: str, model: str):
     nltk.download('punkt')
@@ -17,7 +18,6 @@ def paraphraseTranslateMethod(requestValue: str, model: str):
 
     tokenized_sent_list = sent_tokenize(requestValue)
     result_value = []
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     
     for SENTENCE in tokenized_sent_list:
         if model == 'roen':