Update app.py

app.py

@@ -8,6 +8,9 @@ def load_model(checkpoint_path):
     model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint_path)
     return model, tokenizer
 
+checkpoint_path = "Diezu/Batpho_v2"
+model, tokenizer = load_model(checkpoint_path)
+
 def correct_spelling(text):
     inputs = tokenizer(text, return_tensors="pt", max_length=512, truncation=True)
     outputs = model.generate(**inputs, max_length=512, num_beams=5, early_stopping=True)
@@ -74,4 +77,4 @@ with demo:
     btn_check.click(update_text, inputs=input_text, outputs=[output_html, corrected_textbox, corrected_words_state])
     btn_correct_all.click(apply_full_correction, inputs=[input_text, corrected_words_state], outputs=output_html)
 
-demo.launch()
+demo.launch()