Update app.py

app.py

@@ -15,7 +15,7 @@ model = torch.load("helsinki_fineTuned.pt", map_location=torch.device('cpu'))
 
 
 def translate_gradio(input):
-    encode = model.generate(**tokenizer.prepare_seq2seq_batch(input,return_tensors='pt').to('cuda'))
+    encode = model.generate(**tokenizer.prepare_seq2seq_batch(input,return_tensors='pt'))
     text_ar = tokenizer.batch_decode(encode,skip_special_tokens=True)[0]
     return text_ar