Update app.py

app.py

@@ -1,12 +1,12 @@
 import streamlit as st
 from transformers import RobertaTokenizer, RobertaModel
-from prediction_sinhala.ipynb import MDFEND  # Ensure you import your model correctly
+from prediction_sinhala.py import MDFEND  # Ensure you import your model correctly
 
 # Load model and tokenizer
 @st.cache(allow_output_mutation=True)
 def load_model():
-    tokenizer = RobertaTokenizer.from_pretrained("prediction_sinhala.ipynb")
-    model = MDFEND.from_pretrained("prediction_sinhala.ipynb")
+    tokenizer = RobertaTokenizer.from_pretrained("prediction_sinhala.py")
+    model = MDFEND.from_pretrained("prediction_sinhala.py")
     return model, tokenizer
 
 model, tokenizer = load_model()