Update app.py

app.py

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