Update app.py

app.py

@@ -1,55 +1,20 @@
-# import torch
-# import json
-# import gradio as gr
-
-# # Use a pipeline as a high-level helper
-# from transformers import pipeline
-
-# # model_path = "../Models/models--facebook--nllb-200-distilled-600M/snapshots/f8d333a098d19b4fd9a8b18f94170487ad3f821d"
-
-# # text_translator = pipeline("translation", model=model_path, torch_dtype=torch.bfloat16)
-
-# pipe = pipeline("translation", model="facebook/nllb-200-distilled-600M", torch_dtype=torch.bfloat16)
-
-# # text = "Hello friends, How are you?"
-
-# # translation = text_translator(text, src_lang="eng_Latn", tgt_lang="deu_Latn")
-
-# with open('language.json', 'r') as file:
-#     language_data = json.load(file)
-
-# def get_FLORES_code_from_language(language):
-#     for entry in language_data:
-#         if entry['Language'].lower() == language.lower():
-#             return entry['FLORES-200 code']
-#     return None
+import torch
+import json
+import gradio as gr
 
-# def translate_text(text, destination_language):
-#     dest_code = get_FLORES_code_from_language(destination_language)
-#     translation = text_translator(text, src_lang="eng_Latn", tgt_lang=dest_code)
-#     return translation[0]["translation_text"]
+# Use a pipeline as a high-level helper
+from transformers import pipeline
 
-# gr.close_all()
+# model_path = "../Models/models--facebook--nllb-200-distilled-600M/snapshots/f8d333a098d19b4fd9a8b18f94170487ad3f821d"
 
-# # print(translate_text(text, "Hindi"))
-# demo = gr.Interface(fn=translate_text,
-#                     inputs=[gr.Textbox(label="Input text to translate",lines=6), gr.Dropdown(["German", "French", "Hindi", "Romanian", "Marathi"], label="Select Destination Language")],
-#                     outputs=[gr.Textbox(label="Translated text",lines=4)],
-#                     title="@GenAILearniverse Project 4: Multi language translator",
-#                     description="THIS APPLICATION WILL BE USED TO TRNSLATE ANY ENGLIST TEXT TO MULTIPLE LANGUAGES.")
+# text_translator = pipeline("translation", model=model_path, torch_dtype=torch.bfloat16)
 
-# demo.launch()
+text_translator = pipeline("translation", model="facebook/nllb-200-distilled-600M", torch_dtype=torch.bfloat16)
 
-import torch
-import gradio as gr
-import json
+# text = "Hello friends, How are you?"
 
-# Use a pipeline as a high-level helper
-from transformers import pipeline
+# translation = text_translator(text, src_lang="eng_Latn", tgt_lang="deu_Latn")
 
-text_translator = pipeline("translation", model=model_path,
-                           torch_dtype=torch.bfloat16)
-# Load the JSON data from the file
 with open('language.json', 'r') as file:
     language_data = json.load(file)
 
@@ -59,25 +24,21 @@ def get_FLORES_code_from_language(language):
             return entry['FLORES-200 code']
     return None
 
-
 def translate_text(text, destination_language):
-    # text = "Hello Friends, How are you?"
-    dest_code= get_FLORES_code_from_language(destination_language)
-    translation = text_translator(text,
-                                  src_lang="eng_Latn",
-                                  tgt_lang=dest_code)
+    dest_code = get_FLORES_code_from_language(destination_language)
+    translation = text_translator(text, src_lang="eng_Latn", tgt_lang=dest_code)
     return translation[0]["translation_text"]
 
 gr.close_all()
 
-# demo = gr.Interface(fn=summary, inputs="text",outputs="text")
+# print(translate_text(text, "Hindi"))
 demo = gr.Interface(fn=translate_text,
-                    inputs=[gr.Textbox(label="Input text to translate",lines=6), gr.Dropdown(["German","French", "Hindi", "Romanian	"], label="Select Destination Language")],
+                    inputs=[gr.Textbox(label="Input text to translate",lines=6), gr.Dropdown(["German", "French", "Hindi", "Romanian", "Marathi"], label="Select Destination Language")],
                     outputs=[gr.Textbox(label="Translated text",lines=4)],
                     title="@GenAILearniverse Project 4: Multi language translator",
                     description="THIS APPLICATION WILL BE USED TO TRNSLATE ANY ENGLIST TEXT TO MULTIPLE LANGUAGES.")
-demo.launch()
 
+demo.launch()