Update app.py

app.py

@@ -1,10 +1,10 @@
 # Import the necessary libraries
 import streamlit as st
-from transformers import GPT2Tokenizer, GPT2LMHeadModel, pipeline
+from transformers import GPT2Tokenizer, GPT2Model, pipeline
 import torch
 
 # Load the gpt2-large model and tokenizer for text generation
-gen_model = GPT2LMHeadModel.from_pretrained('gpt2-large')
+gen_model = GPT2Model.from_pretrained('gpt2-large')
 gen_tokenizer = GPT2Tokenizer.from_pretrained('gpt2-large')
 
 # Load the zero-shot text classification pipeline from HuggingFace