Akbartus commited on
Commit
21627f8
·
1 Parent(s): f83b9fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -24,12 +24,7 @@ def shot(image, labels_text):
24
 
25
 
26
  # Translate
27
- tokenizer.src_lang = "en"
28
- encodedText = tokenizer(label, return_tensors="pt")
29
- generatedTokens = model.generate(**encodedText, forced_bos_token_id=tokenizer.get_lang_id("ru"))
30
- return tokenizer.batch_decode(generatedTokens, skip_special_tokens=True)[0]
31
 
32
-
33
  res = pipe(images=PIL_image,
34
  candidate_labels=labels,
35
  hypothesis_template= "This is a photo of a {}")
 
24
 
25
 
26
  # Translate
 
 
 
 
27
 
 
28
  res = pipe(images=PIL_image,
29
  candidate_labels=labels,
30
  hypothesis_template= "This is a photo of a {}")