Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import pandas as pd
 import os
 import torch
-from transformers import GPT2LMHeadModel, GPT2Tokenizer, AutoTokenizer, AutoModelForCausalLM, pipeline
+from transformers import GPT2Model, GPT2Tokenizer, AutoTokenizer, AutoModelForCausalLM, pipeline
 from huggingface_hub import HfFolder
 from io import StringIO
 
@@ -18,7 +18,7 @@ os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
 
 # Load the tokenizer and model
 tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
-model_gpt2 = GPT2LMHeadModel.from_pretrained('gpt2')
+model_gpt2 = GPT2Model.from_pretrained('gpt2')
 
 # Create a pipeline for text generation using GPT-2
 text_generator = pipeline("text-generation", model=model_gpt2, tokenizer=tokenizer)
@@ -34,7 +34,7 @@ def load_llama_model():
         model_llama = AutoModelForCausalLM.from_pretrained(
             model_name,
             torch_dtype=torch.float16,  # Use FP16 for reduced memory
-            use_auth_token=hf_token
+            token=hf_token
         )
         tokenizer_llama = AutoTokenizer.from_pretrained(model_name, token=hf_token)