Update app.py

app.py

@@ -2,13 +2,13 @@ import os
 os.system('pip install streamlit transformers torch')
 
 import streamlit as st
-from transformers import BartForConditionalGeneration,   AutoTokenizer
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch
 
-# Load the BART fine tuned model and tokenizer
-model_name = "llmahmad/facebook_BART_summary"
-tokenizer = BartForConditionalGeneration.from_pretrained(model_name)
-model =  AutoTokenizer.from_pretrained(model_name)
+# Load the GPT-2 model and tokenizer
+model_name = 'gpt2-large'
+tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+model = GPT2LMHeadModel.from_pretrained(model_name)
 
 def generate_blog_post(topic):
     try: