Update src/streamlit_app.py

src/streamlit_app.py

@@ -26,27 +26,29 @@ hf_token = os.getenv("HF_TOKEN")
 # use cache
 @st.cache_resource
 def load_tokenizer(model_path):
+
+    Try:
+    
+        if "wangchanberta" in model_path.lower():
+            return AutoTokenizer.from_pretrained(
+                model_path, 
+                token=hf_token, 
+                use_fast=False,
+                trust_remote_code=True
+            )
+    
+        tokenizer = AutoTokenizer.from_pretrained(
+                model_path, 
+                token=hf_token, 
+                trust_remote_code=True
+            )
     
-    if "wangchanberta" in model_path.lower():
-        return AutoTokenizer.from_pretrained(
-            model_path, 
-            token=hf_token, 
-            use_fast=False,
-            trust_remote_code=True
-        )
-
-    tokenizer = AutoTokenizer.from_pretrained(
-            model_path, 
-            token=hf_token, 
-            trust_remote_code=True
-        )
-
-    if tokenizer.pad_token is None:
-        tokenizer.pad_token = tokenizer.eos_token
+        if tokenizer.pad_token is None:
+            tokenizer.pad_token = tokenizer.eos_token
+        
         return tokenizer
 
     except Exception as e:
-        # ส่ง Error กลับไปแสดงที่หน้า UI
         st.error(f"Error loading tokenizer for {model_path}: {e}")
         return None