Update app.py

app.py

@@ -5,6 +5,7 @@ from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, ServiceCon
 from llama_index.llms.huggingface import HuggingFaceLLM as LlamaHuggingFaceLLM
 from llama_index.core.prompts.prompts import SimpleInputPrompt
 from llama_index.legacy.embeddings.langchain import LangchainEmbedding
+import torch
 
 # Setup for caching the index and LLM to avoid reloading
 @st.cache(allow_output_mutation=True, suppress_st_warning=True)