Spaces:

vanhai123
/

VietDescriberAI

Sleeping

vanhai123 commited on May 26

Commit

7588485

verified ·

1 Parent(s): 16aefd2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,9 @@ blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image
 translator_tokenizer = NllbTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
 translator_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
 lang_code = "vie_Latn"
-lang_token_id = translator_tokenizer.lang_code_to_id[lang_code]
 def caption_translate(image):
     # Step 1: Get English caption

 translator_tokenizer = NllbTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
 translator_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
 lang_code = "vie_Latn"
+# Manually map lang_code to token ID (based on tokenizer's config)
+lang_token_id = translator_tokenizer.convert_tokens_to_ids(lang_code)
 def caption_translate(image):
     # Step 1: Get English caption