Akbartus commited on
Commit
c5fe961
·
1 Parent(s): 94e11e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,8 +25,8 @@ def shot(image, labels_text):
25
  hypothesis_template= "This is a photo of a {}")
26
  return {dic["label"]: dic["score"] for dic in res}
27
  # Translate
28
- tokenizer.src_lang = "en"
29
- encodedText = tokenizer(labels_text, return_tensors="pt")
30
  generatedTokens = model.generate(**encodedText, forced_bos_token_id=tokenizer.get_lang_id("ru"))
31
  return tokenizer.batch_decode(generatedTokens, skip_special_tokens=True)[0]
32
 
 
25
  hypothesis_template= "This is a photo of a {}")
26
  return {dic["label"]: dic["score"] for dic in res}
27
  # Translate
28
+ tokenizer.src_lang = "en"
29
+ encodedText = tokenizer(labels_text, return_tensors="pt")
30
  generatedTokens = model.generate(**encodedText, forced_bos_token_id=tokenizer.get_lang_id("ru"))
31
  return tokenizer.batch_decode(generatedTokens, skip_special_tokens=True)[0]
32