Amandeep01 commited on
Commit
57b25d2
·
verified ·
1 Parent(s): af59cc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -16,10 +16,10 @@ model = MarianMTModel.from_pretrained(model_name)
16
  # Font path (this works on Hugging Face Spaces)
17
  FONT_PATH = "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf"
18
 
19
- # Translation function
20
  def translate_text(text):
21
- batch = tokenizer.prepare_seq2seq_batch([text], return_tensors="pt")
22
- translated = model.generate(**batch)
23
  return tokenizer.decode(translated[0], skip_special_tokens=True)
24
 
25
  # Main function
 
16
  # Font path (this works on Hugging Face Spaces)
17
  FONT_PATH = "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf"
18
 
19
+ # Translation function (cleaned and CPU safe)
20
  def translate_text(text):
21
+ inputs = tokenizer(text, return_tensors="pt")
22
+ translated = model.generate(**inputs)
23
  return tokenizer.decode(translated[0], skip_special_tokens=True)
24
 
25
  # Main function