use llama model

app.py

@@ -9,6 +9,8 @@ import planet_model as customModel
 from typing import List, Dict
 import constants
 import os
+from ollama import chat
+from ollama import ChatResponse
 
 api_key = os.getenv("api_key")
 
@@ -86,6 +88,15 @@ async def main():
         duration = ""
         purpose = ""
         interests = ""
+    response: ChatResponse = chat(model='llama3.2', messages=[
+        {
+            'role': 'user',
+            'content': 'Why is the sky blue?',
+        },
+            ])
+    print(response['message']['content'])
+    # or access fields directly from the response object
+    print(response.message.content)
         
 
 if __name__ == '__main__':