Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
-from llama_index import StorageContext, load_index_from_storage, VectorStoreIndex, SimpleDirectoryReader, Query
+from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, StorageContext
 from llama_index.llms.huggingface import HuggingFaceInferenceAPI
-from dotenv import load_dotenv
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
+from dotenv import load_dotenv
 from llama_index.core import Settings
 import os
 import base64
@@ -57,7 +57,7 @@ def data_ingestion():
 
 def handle_query(query):
     storage_context = StorageContext.from_defaults(persist_dir=PERSIST_DIR)
-    index = load_index_from_storage(storage_context)
+    index = VectorStoreIndex.from_storage_context(storage_context)
     
     results = index.search(query)