Update app.py

app.py

@@ -5,9 +5,12 @@ from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain, SequentialChain
 from langchain.memory import ConversationBufferMemory
 from langchain.utilities import WikipediaAPIWrapper
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
-OPENAI_KEY = os.getenv("OPENAI_API_KEY")
-os.environ['OPENAI_API_KEY'] = OPENAI_KEY
+# Load the GPT-2 model and tokenizer
+model_name = "gpt2"
+tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+OpenAI = GPT2LMHeadModel.from_pretrained(model_name)
 
 
 # app framework
@@ -34,7 +37,7 @@ script_chain=LLMChain(llm=llm,prompt=script_template,verbose=True,output_key='sc
 
 wiki=WikipediaAPIWrapper()
 
-# Show stuff to teh screen if there is a prompt
+# Show stuff to the screen if there is a prompt
 if prompt:
     title=title_chain.run(prompt)
     wiki_research=wiki.run(prompt)