Update docs

Author: DhruvSrikanth

README.md

@@ -1,7 +1,7 @@
 <div align="center" style="display: flex; align-items: center; justify-content: center;">
   <picture>
     <source srcset="docs/assets/ai_function_light.png" media="(prefers-color-scheme: dark)">
-    <img src="docs/assets/ai_function_dark.png" alt="Your Image" style="max-width: 100%;">
+    <img src="docs/assets/ai_function_dark.png" alt="AI Function" style="max-width: 100%;">
   </picture>
 </div>
 

docs/assets/ai_function_dark.png

@@ -212,8 +212,3 @@ output, metadata = task_evaluator("I want to train a model to predict house pric
 for key, value in output.items(): print(f"{key}: {value}")
 for i, step in enumerate(metadata["reasoning_steps"]): print(f"Step {i+1}: {step}")
 ```
-
-
-```python
-
-```

docs/assets/ai_function_light.png

@@ -354,13 +354,6 @@
     "for key, value in output.items(): print(f\"{key}: {value}\")\n",
     "for i, step in enumerate(metadata[\"reasoning_steps\"]): print(f\"Step {i+1}: {step}\")"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {