Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,11 +1,21 @@
 
 # ONCE
-from transformers import GPT2TokenizerFast, GPT2Model
-import os
+# from transformers import GPT2TokenizerFast, GPT2Model
+# import os
+
+# # Load from local offline folder
+# model = GPT2Model.from_pretrained("./models")
+# tokenizer = GPT2TokenizerFast.from_pretrained("./models")
+
+from transformers import GPT2Model, GPT2TokenizerFast
+
+model = GPT2Model.from_pretrained("gpt2")
+tokenizer = GPT2TokenizerFast.from_pretrained("gpt2")
+
+model.save_pretrained("./models")
+tokenizer.save_pretrained("./models")
 
-# Load from local offline folder
-model = GPT2Model.from_pretrained("./models")
-tokenizer = GPT2TokenizerFast.from_pretrained("./models")
+print("✅ Downloaded and saved GPT-2 to ./models")