Update app.py

app.py

@@ -23,18 +23,23 @@ def shot(image, labels_text):
     labels = labels_text.split(",")
     
    
-
+    # 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]
+    
 
     res = pipe(images=PIL_image, 
            candidate_labels=labels,
            hypothesis_template= "This is a photo of a {}")
-    return {dic["label"]: dic["score"] for dic in res}
-     # Translate 
+    return {dic["label"]: dic["score"] for dic in res
     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]
-    
+    }
+     
     
 iface = gr.Interface(shot, 
                     ["image", "text"],