Spaces:

dave21-py
/

tiny_siri_demo

Running

dave21-py commited on Nov 28, 2025

Commit

b5e286a

verified ·

1 Parent(s): e318aa9

Update streamlit_app.py

Files changed (1) hide show

streamlit_app.py CHANGED Viewed

@@ -19,7 +19,7 @@ def load_assets():
     tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-uncased')
     # Load the Tiny Model (Explicitly set map_location and weights_only=False)
-    model = torch.load("siri_bert_quantized.pt", map_location=torch.device('cpu'))
     model.eval()
     return tokenizer, model, encoder

     tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-uncased')
     # Load the Tiny Model (Explicitly set map_location and weights_only=False)
+    model = torch.load("siri_bert_quantized.pt", map_location=torch.device('cpu'), weights_only=False)
     model.eval()
     return tokenizer, model, encoder