Update main.py

main.py

@@ -50,7 +50,11 @@ async def reply_whatsapp(request: Request):
         global query_engine,index
         storage_context = StorageContext.from_defaults(persist_dir="llama_index")
         index = load_index_from_storage(storage_context=storage_context)
-        query_engine = index.as_query_engine()
+        
+        # Set the GPT model (e.g., GPT-4, GPT-3.5 Turbo)
+        llm = OpenAI(model="gpt-4o-mini", api_key=os.environ["OPENAI_API_KEY"])  # Use "gpt-4" if desired
+
+        query_engine = index.as_query_engine(llm=llm)
         print("loaded")
         
         gpt_response = query_engine.query(f"""