Upload assignment_8.py

assignment_8.py

@@ -0,0 +1,25 @@
+import gradio as gr
+from transformers import MarianTokenizer, MarianMTModel
+
+# Load the translation model and tokenizer
+model_name = "Helsinki-NLP/opus-mt-en-ro"
+tokenizer = MarianTokenizer.from_pretrained(model_name)
+model = MarianMTModel.from_pretrained(model_name)
+
+def translate(text):
+    # Tokenize and translate the input text
+    inputs = tokenizer(text, return_tensors="pt")
+    translation = model.generate(**inputs)
+    translated_text = tokenizer.decode(translation[0], skip_special_tokens=True)
+    return translated_text
+
+# Create a Gradio interface
+app = gr.Interface(
+    fn=translate,
+    inputs=gr.Textbox(label="Enter an English sentence:"),
+    outputs=gr.Textbox(label="Romanian Translation:"),
+    live=True,
+)
+
+if __name__ == "__main__":
+    app.launch()