Spaces:

Akbartus
/

ZeroShot-Image-Classification-Experimental

Runtime error

Akbartus commited on Oct 14, 2022

Commit

d5aaaf8

1 Parent(s): ed80744

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ def shot(image, labels_text):
     # Translate
     tokenizer.src_lang = "en"
-    encodedText = tokenizer(labels_text, return_tensors="pt")
     generatedTokens = model.generate(**encodedText, forced_bos_token_id=tokenizer.get_lang_id("ru"))
     return tokenizer.batch_decode(generatedTokens, skip_special_tokens=True)[0]

     # Translate
     tokenizer.src_lang = "en"
+    encodedText = tokenizer(label, return_tensors="pt")
     generatedTokens = model.generate(**encodedText, forced_bos_token_id=tokenizer.get_lang_id("ru"))
     return tokenizer.batch_decode(generatedTokens, skip_special_tokens=True)[0]