Update app.py

app.py

@@ -19,8 +19,6 @@ TOKEN = os.environ['TOKEN']
 # Translation function for Nano and Large models
 @spaces.GPU
 def translate_nano_large(text, model_path):
-    device = "cuda:0" if torch.cuda.is_available() else "cpu"
-    print(f'[INFO] Using device: {device}')
     translator = pipeline("translation", model=model_path, token=TOKEN)
     translated = translator(
         text, 
@@ -32,7 +30,6 @@ def translate_nano_large(text, model_path):
         pad_token_id=translator.tokenizer.pad_token_id,
         bos_token_id=translator.tokenizer.bos_token_id,
         eos_token_id=translator.tokenizer.eos_token_id,
-        device=device,
     )
     return translated[0]["translation_text"]