Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSeq2SeqLM
-from llama_index import VectorStoreIndex, SimpleDirectoryReader, ServiceContext
-from llama_index.llms import HuggingFaceLLM
-from llama_index.embeddings import HuggingFaceEmbedding
+from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, ServiceContext
+from llama_index.llms.huggingface import HuggingFaceLLM
+from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 import os
 
 # Load DeepSeek-R1
@@ -15,7 +15,14 @@ indicbart_model = AutoModelForSeq2SeqLM.from_pretrained("ai4bharat/IndicBART")
 
 # Initialize LlamaIndex components
 embed_model = HuggingFaceEmbedding(model_name="sentence-transformers/all-MiniLM-L6-v2")
-llm = HuggingFaceLLM(model=deepseek_model, tokenizer=deepseek_tokenizer)
+
+llm = HuggingFaceLLM(
+    context_window=4096,
+    max_new_tokens=512,
+    model="deepseek-ai/DeepSeek-R1",
+    tokenizer_name="deepseek-ai/DeepSeek-R1"
+)
+
 service_context = ServiceContext.from_defaults(llm=llm, embed_model=embed_model)
 
 # Build index from documents in 'data' directory