Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ def shot(image, labels_text):
|
|
35 |
hypothesis_template= "This is a photo of a {}")
|
36 |
|
37 |
tokenizer.src_lang = "en"
|
38 |
-
encodedText = tokenizer(
|
39 |
generatedTokens = model.generate(**encodedText, forced_bos_token_id=tokenizer.get_lang_id("ru"))
|
40 |
return tokenizer.batch_decode(generatedTokens, skip_special_tokens=True)[0], {dic["label"]: dic["score"] for dic in res}
|
41 |
|
|
|
35 |
hypothesis_template= "This is a photo of a {}")
|
36 |
|
37 |
tokenizer.src_lang = "en"
|
38 |
+
encodedText = tokenizer(labels_text, return_tensors="pt")
|
39 |
generatedTokens = model.generate(**encodedText, forced_bos_token_id=tokenizer.get_lang_id("ru"))
|
40 |
return tokenizer.batch_decode(generatedTokens, skip_special_tokens=True)[0], {dic["label"]: dic["score"] for dic in res}
|
41 |
|