Update app.py

app.py

@@ -1,9 +1,12 @@
 import streamlit as st
 from transformers import MarianTokenizer, MarianMTModel , BertTokenizer, AutoModelForSeq2SeqLM, pipeline
-from ar_correct import ar_correct
+from ar_corrector.corrector import Corrector
+
 import mishkal.tashkeel 
+
 from arabert.preprocess import ArabertPreprocessor
 
+
 # Initialize Mishkal vocalizer
 vocalizer = mishkal.tashkeel.TashkeelClass()
 
@@ -19,7 +22,7 @@ tokenizer_summarization = BertTokenizer.from_pretrained(model_name)
 model_summarization = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 pipeline_summarization = pipeline("text2text-generation", model=model_summarization, tokenizer=tokenizer_summarization)
 
-
+corr = Corrector()
 def main():
     st.title("U3reb Demo")
 
@@ -42,7 +45,7 @@ def main():
     # Arabic Text Correction
     st.subheader("Arabic Text Correction (ar_correct)")
     if input_text:
-        corrected_text = ar_correct(input_text)
+        corrected_text = corr.contextual_correct(input_text)
         st.write("Corrected Text:", corrected_text)
 
     # Text Summarization