Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,3 +1,14 @@
+
+# ONCE
+from transformers import GPT2TokenizerFast, GPT2Model
+import os
+
+# Load from local offline folder
+model = GPT2Model.from_pretrained("./models")
+tokenizer = GPT2TokenizerFast.from_pretrained("./models")
+
+
+
 import streamlit as st
 st.set_page_config(page_title="GPT-2 Attention Explorer", layout="wide")