Update app/agents/crew_pipeline.py

app/agents/crew_pipeline.py

@@ -79,7 +79,8 @@ translation_model = AutoModelForSeq2SeqLM.from_pretrained(
     device_map = 'auto' if DEVICE == 'cuda' else None
     
 )
-
+if DEVICE == 'cpu':
+    translation_model = translation_model.to('cpu')
 
 
     
@@ -126,7 +127,7 @@ def translate_text(text: str, src_lang: str, tgt_lang: str, max_chunk_len: int =
             padding = True,
             truncation = True,
             max_length = 400 
-        ).to(translation_model,device)
+        ).to(translation_model.device)
 
         #setting the target language token
         forced_bos_token_id = translation_tokenizer.convert_tokens_to_ids(tgt_lang)