Update app.py

app.py

@@ -5,6 +5,7 @@ from langchain.chains.conversation.memory import ConversationBufferWindowMemory
 from langchain.llms import HuggingFacePipeline
 from langchain import PromptTemplate
 from typing import List
+import torch
 # Load the model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("gpt2")
 model = AutoModelForCausalLM.from_pretrained("gpt2")