Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,15 +7,20 @@
 # model = GPT2Model.from_pretrained("./models")
 # tokenizer = GPT2TokenizerFast.from_pretrained("./models")
 
-from transformers import GPT2Model, GPT2TokenizerFast
+# from transformers import GPT2Model, GPT2TokenizerFast
 
-model = GPT2Model.from_pretrained("gpt2")
-tokenizer = GPT2TokenizerFast.from_pretrained("gpt2")
+# model = GPT2Model.from_pretrained("gpt2")
+# tokenizer = GPT2TokenizerFast.from_pretrained("gpt2")
+
+# model.save_pretrained("./models")
+# tokenizer.save_pretrained("./models")
 
-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")
 
-print("✅ Downloaded and saved GPT-2 to ./models")
+print("✅ Downloaded and saved GPT-2 to models")
 
 
 
@@ -31,8 +36,8 @@ import pandas as pd
 
 @st.cache_resource
 def load_model():
-    tokenizer = GPT2TokenizerFast.from_pretrained("./models")
-    model = GPT2Model.from_pretrained("./models", output_attentions=True, attn_implementation="eager")
+    tokenizer = GPT2TokenizerFast.from_pretrained("models")
+    model = GPT2Model.from_pretrained("models", output_attentions=True, attn_implementation="eager")
     model.eval()
     return tokenizer, model