menikev commited on
Commit
2905417
·
verified ·
1 Parent(s): 78e2935

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import streamlit as st
2
  from transformers import RobertaTokenizer, RobertaModel
3
- from your_model_file import MDFEND # Ensure you import your model correctly
4
 
5
  # Load model and tokenizer
6
  @st.cache(allow_output_mutation=True)
 
1
  import streamlit as st
2
  from transformers import RobertaTokenizer, RobertaModel
3
+ from prediction_sinhala.ipynb import MDFEND # Ensure you import your model correctly
4
 
5
  # Load model and tokenizer
6
  @st.cache(allow_output_mutation=True)