Update app.py

app.py

@@ -1,36 +1,21 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline, AutoModelForSeq2SeqLM
+from transformers import pipeline
 
-# Load the keyphrase extraction models
-tokenizer1 = AutoTokenizer.from_pretrained("ml6team/keyphrase-extraction-kbir-inspec")
-model1 = AutoModelForTokenClassification.from_pretrained("ml6team/keyphrase-extraction-kbir-inspec")
-extractor1 = pipeline("ner", model=model1, tokenizer=tokenizer1)
-
-tokenizer2 = AutoTokenizer.from_pretrained("aglazkova/bart_finetuned_keyphrase_extraction")
-model2 = AutoModelForSeq2SeqLM.from_pretrained("aglazkova/bart_finetuned_keyphrase_extraction")
-extractor2 = pipeline("text2text-generation", model=model2, tokenizer=tokenizer2)
+# Initialize the translation pipeline
+translator = pipeline("translation_ro_to_en")
 
 # Streamlit app
-st.title("Keyphrase Extraction App")
-st.write("Enter text below to extract keyphrases using two different models:")
-
-# Text input
-user_input = st.text_area("Text Input", "")
+st.title("Romanian to English Translator")
 
-# Extract keyphrases
-if st.button("Extract Keyphrases"):
-    if user_input:
-        # Extract keyphrases using the first model
-        keyphrases_model1 = extractor1(user_input)
-        st.write("Extracted Keyphrases from `ml6team/keyphrase-extraction-kbir-inspec`:")
-        for keyphrase in keyphrases_model1:
-            st.write(keyphrase['word'])
+# Input text
+input_text = st.text_area("Enter text in Romanian:")
 
-        # Extract keyphrases using the second model
-        tokenized_text = tokenizer2.prepare_seq2seq_batch([user_input], return_tensors='pt')
-        translation = model2.generate(**tokenized_text)
-        translated_text = tokenizer2.batch_decode(translation, skip_special_tokens=True)[0]
-        st.write("Extracted Keyphrases from `aglazkova/bart_finetuned_keyphrase_extraction`:")
-        st.write(translated_text)
+# Translate button
+if st.button("Translate"):
+    if input_text:
+        # Perform translation
+        translation = translator(input_text)[0]['translation_text']
+        st.write("Translation:")
+        st.write(translation)
     else:
-        st.write("Please enter some text to extract keyphrases.")
+        st.write("Please enter some text to translate.")