Update app.py

app.py

@@ -14,10 +14,10 @@ from tqdm import tqdm
 import numpy as np
 import pandas as pd
 kv = KhaveeVerifier()
-from transformers import AutoTokenizer, TFAutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("Thanravee/KarveeSaimai", local_files_only=False)
-model = TFAutoModelForCausalLM.from_pretrained("Thanravee/KarveeSaimai", local_files_only=False)
+model = AutoModelForCausalLM.from_pretrained("Thanravee/KarveeSaimai", local_files_only=False)
 
 # split text from \n to list and drop soi word ->  splitted wak list (no soi)
 def split_klong(klong_text):