Update app.py

app.py

@@ -1,10 +1,13 @@
 import streamlit as st
 import torch
+import torch.serialization
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
 @st.cache_resource
-def load_model():
-    checkpoint = torch.load('TinyBERT_cls_model.pt', map_location='cpu')
+def load_model():   
+    checkpoint = torch.load('TinyBERT_cls_model.pt', 
+                          map_location='cpu',
+                          weights_only=False)  
     
     model = AutoModelForSequenceClassification.from_pretrained(
         "huawei-noah/TinyBERT_General_4L_312D",