Spaces:

SeyedAli
/

Persian-Image-Captioning-1

Sleeping

SeyedAli commited on Sep 24, 2023

Commit

a1b73ef

•

1 Parent(s): eaa52e3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,8 +6,7 @@ from PIL import Image
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model_size = "small"
-model_name = f"persiannlp/mt5-{model_size}-parsinlu-translation_en_fa"
 translation_tokenizer = MT5Tokenizer.from_pretrained(model_name)
 translation_model = MT5ForConditionalGeneration.from_pretrained(model_name)

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model_name = "SeyedAli/English-to-Persian-Translation-mT5-V1"
 translation_tokenizer = MT5Tokenizer.from_pretrained(model_name)
 translation_model = MT5ForConditionalGeneration.from_pretrained(model_name)