Update app.py

app.py

@@ -9,8 +9,13 @@ import gradio as gr
 
 
 
-model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
-tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M")
+#model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
+#tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M")
+#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]
+    
 
 pipe = pipeline("zero-shot-image-classification", model="openai/clip-vit-base-patch32")
 images="dog.jpg"
@@ -29,10 +34,7 @@ def shot(image, labels_text):
            candidate_labels=labels,
            hypothesis_template= "This is a photo of a {}")
     
-    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], {dic["label"]: dic["score"] for dic in res}
+    return {dic["label"]: dic["score"] for dic in res}
      
     
 iface = gr.Interface(shot,