Spaces:

wangjin2000
/

Comparing_Image_Captions

Paused

wangjin2000 commited on May 31, 2023

Commit

9b0261c

1 Parent(s): 3ec479a

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",model=""Helsinki-NLP/opus-mt-zh-en")
 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)