Update app.py

app.py

@@ -8,18 +8,18 @@ import gradio as gr
 
 tsr.pytesseract.tesseract_cmd = r'/usr/bin/tesseract'
 
-model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_1.2B")
-tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_1.2B")
+model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
+tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M")
 
 def extractAndTranslate(image):
     # Extract Text
-    extractedText = tsr.image_to_string(image, lang='eng+uzb')
+    extractedText = tsr.image_to_string(image, lang='eng+rus')
     extractedTextFormatted = ' '.join(extractedText.split('\n'))
     
     # Translate 
     tokenizer.src_lang = "en"
     encodedText = tokenizer(extractedTextFormatted, return_tensors="pt")
-    generatedTokens = model.generate(**encodedText, forced_bos_token_id=tokenizer.get_lang_id("uz"))
+    generatedTokens = model.generate(**encodedText, forced_bos_token_id=tokenizer.get_lang_id("ru"))
     
     return tokenizer.batch_decode(generatedTokens, skip_special_tokens=True)[0]