Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSeq2SeqLM
 from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, ServiceContext
-from llama_index.llms.huggingface import HuggingFaceLLM
+from llama_index.llms import HuggingFaceLLM
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 import os
 
@@ -17,10 +17,10 @@ indicbart_model = AutoModelForSeq2SeqLM.from_pretrained("ai4bharat/IndicBART")
 embed_model = HuggingFaceEmbedding(model_name="sentence-transformers/all-MiniLM-L6-v2")
 
 llm = HuggingFaceLLM(
-    context_window=4096,
+    model_name="deepseek-ai/DeepSeek-R1",
+    tokenizer_name="deepseek-ai/DeepSeek-R1",
     max_new_tokens=512,
-    model="deepseek-ai/DeepSeek-R1",
-    tokenizer_name="deepseek-ai/DeepSeek-R1"
+    context_window=4096
 )
 
 service_context = ServiceContext.from_defaults(llm=llm, embed_model=embed_model)