Update app.py

app.py

@@ -1,8 +1,16 @@
+import os
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, GPT2LMHeadModel, GPT2Tokenizer
 
-# Load your Hugging Face model
-generator = pipeline('text-generation', model='Jonathanmann/GPT2-medium-SADnov21')
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+# Load the tokenizer and model explicitly as GPT2 to avoid config issues
+model_name = 'Jonathanmann/GPT2-medium-SADnov21'
+tokenizer = GPT2Tokenizer.from_pretrained(model_name, use_auth_token=HF_TOKEN)
+model = GPT2LMHeadModel.from_pretrained(model_name, use_auth_token=HF_TOKEN)
+
+# Define the text generation pipeline
+generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
 
 # Define a function for generating text
 def generate_text(prompt, max_length, temperature, top_k, top_p):