Update app.py

app.py

@@ -3,8 +3,8 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
 @st.cache(allow_output_mutation=True)
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("./my_model")
-    model = AutoModelForSequenceClassification.from_pretrained("./my_model")
+    tokenizer = AutoTokenizer.from_pretrained("./prediction_sinhala.ipynb")
+    model = AutoModelForSequenceClassification.from_pretrained("./prediction_sinhala.ipynb")
     return model, tokenizer
 
 model, tokenizer = load_model()