Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -15,7 +15,7 @@ pipe = pipeline("zero-shot-image-classification", model="openai/clip-vit-base-pa
|
|
| 15 |
images="dog.jpg"
|
| 16 |
|
| 17 |
def shot(image, labels_text):
|
| 18 |
-
|
| 19 |
|
| 20 |
|
| 21 |
PIL_image = Image.fromarray(np.uint8(image)).convert('RGB')
|
|
@@ -25,7 +25,7 @@ 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 |
-
|
| 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]
|
|
|
|
| 15 |
images="dog.jpg"
|
| 16 |
|
| 17 |
def shot(image, labels_text):
|
| 18 |
+
|
| 19 |
|
| 20 |
|
| 21 |
PIL_image = Image.fromarray(np.uint8(image)).convert('RGB')
|
|
|
|
| 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]
|