Update src/streamlit_app.py

src/streamlit_app.py

@@ -15,10 +15,17 @@
 # model.save_pretrained("./models")
 # tokenizer.save_pretrained("./models")
 
-model = GPT2Model.from_pretrained("openai-community/gpt2")
-tokenizer = GPT2TokenizerFast.from_pretrained("openai-community/gpt2")
-model.save_pretrained("models")
-tokenizer.save_pretrained("models")
+# model = GPT2Model.from_pretrained("openai-community/gpt2")
+# tokenizer = GPT2TokenizerFast.from_pretrained("openai-community/gpt2")
+# model.save_pretrained("models")
+# tokenizer.save_pretrained("models")
+
+
+from transformers import GPT2Tokenizer, GPT2Model
+
+model_id = "gpt2"
+GPT2Model.from_pretrained(model_id).save_pretrained("models")
+GPT2Tokenizer.from_pretrained(model_id).save_pretrained("models")
 
 print("✅ Downloaded and saved GPT-2 to models")