Comparing_Image_Captions

Runtime error

wangjin2000 commited on May 31, 2023

Commit

3ec479a

•

1 Parent(s): 672c8e9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ checkpoint2 = "wangjin2000/git-base-finetune"
 #model2 = AutoModelForCausalLM.from_pretrained(checkpoint2, use_auth_token=access_token)
 model2 = AutoModelForCausalLM.from_pretrained(checkpoint2)
-en_zh_translator = pipeline("translation",src_lang = "en", tgt_lang = "zh")
 def img2cap_com(image):
     input1 = processor(images=image, return_tensors="pt").to(device)

 #model2 = AutoModelForCausalLM.from_pretrained(checkpoint2, use_auth_token=access_token)
 model2 = AutoModelForCausalLM.from_pretrained(checkpoint2)
+en_zh_translator = pipeline("translation",model=""Helsinki-NLP/opus-mt-zh-en")
 def img2cap_com(image):
     input1 = processor(images=image, return_tensors="pt").to(device)