Update app.py

app.py

@@ -1,10 +1,8 @@
-pip install --upgrade transformers
-
 import streamlit as st
 import pandas as pd
 import os
 import torch
-from transformers import GPT2Model, GPT2Tokenizer, AutoTokenizer, AutoModelForCausalLM, pipeline
+from transformers import GPT2LMHeadModel, GPT2Tokenizer, AutoTokenizer, AutoModelForCausalLM, pipeline
 from huggingface_hub import HfFolder
 from io import StringIO
 
@@ -20,7 +18,7 @@ os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
 
 # Load the tokenizer and model
 tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
-model_gpt2 = GPT2Model.from_pretrained('gpt2')
+model_gpt2 = GPT2LMHeadModel.from_pretrained('gpt2')
 
 # Create a pipeline for text generation using GPT-2
 text_generator = pipeline("text-generation", model=model_gpt2, tokenizer=tokenizer)