Update app.py

app.py

@@ -9,8 +9,15 @@ hf_token = os.getenv("gpt2_token")
 
 if not hf_token:
     raise ValueError("Hugging Face token not found. Please set HF_TOKEN as an environment variable.")
+
+model = AutoModelForCausalLM.from_pretrained(
+    "isitcoding/gpt2_120_finetuned",
+    config="adapter_config.json",  # Specify the custom config file
+    state_dict="adapter_model.safetensors"  # Specify the custom model weights
+)
+tokenizer = AutoTokenizer.from_pretrained("isitcoding/gpt2_120_finetuned")
 # Load the text generation pipeline with your fine-tuned model
-generator = pipeline('text-generation', model='isitcoding/gpt2_120_finetuned', use_auth_token = hf_token)
+generator = pipeline('text-generation', model=model, tokenizer=tokenizer, use_auth_token = hf_token)
 
 # Function to generate responses using the text generation model
 def respond(message, chat_history):