Update app.py

app.py

@@ -15,6 +15,11 @@ pipe = pipeline("zero-shot-image-classification", model="openai/clip-vit-base-pa
 images="dog.jpg"
 
 def shot(image, labels_text):
+ 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]
+    
     PIL_image = Image.fromarray(np.uint8(image)).convert('RGB')
     labels = labels_text.split(",")
     res = pipe(images=PIL_image, 
@@ -22,11 +27,7 @@ def shot(image, labels_text):
            hypothesis_template= "This is a photo of a {}")
     return {dic["label"]: dic["score"] for dic in res}
         # Translate 
-    tokenizer.src_lang = "en"
-    encodedText = tokenizer(candidate_labels, 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]
+