Update app.py

app.py

@@ -93,7 +93,7 @@ def generate_image(image, product_name, target_name):
 def translate_sentence(article, source, target):
     if target == 'eng_Latn':
       return article
-    translator = pipeline('translation', model=trans_model, tokenizer=trans_tokenizer, src_lang=source, tgt_lang=target, device=0)
+    translator = pipeline('translation', model=trans_model, tokenizer=trans_tokenizer, src_lang=source, tgt_lang=target)
     output = translator(article, max_length=400)
     output = output[0]['translation_text']
     return output