Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 {}")
|