Update README.md

README.md

@@ -9,27 +9,30 @@ license: mit
 ---
 
 
+# 🤖 Ollama AI Assistant
 
-title: Ollama AI Assistant emoji: 🤖 colorFrom: purple colorTo: blue sdk: --- docker app_port: 8501 --- 
-Ollama AI Assistant
-This Hugging Face Space hosts an AI assistant powered by Ollama, FastAPI, and Streamlit.
+This project hosts a lightweight AI assistant powered by **Ollama**, **FastAPI**, and **Streamlit**, all bundled in a single Docker environment.
 
-Ollama: The large language model (LLM) inference engine.
+## 🚀 Overview
 
-FastAPI: Provides a robust backend API for interacting with the LLM.
+- **Ollama** – Runs and serves the LLM model.
+- **FastAPI** – Handles backend API requests to interact with the model.
+- **Streamlit** – Provides a user-friendly web UI.
+- **Docker** – Runs everything in isolated and reproducible containers.
 
-Streamlit: Offers an intuitive and interactive web-based user interface.
+---
 
-How it Works
-The Docker container starts, running Ollama, FastAPI, and Streamlit concurrently.
+## 🧠 How It Works
 
-Ollama pulls and serves the krishna_choudhary/lightweight_chatbot model.
+1. **Ollama** loads the LLM model: `krishna_choudhary/lightweight_chatbot`.
+2. **FastAPI** provides an API backend (running on internal port `7860`) for prompt-response communication.
+3. **Streamlit UI** (exposed on port `8501`) lets users enter prompts and receive responses.
+4. The UI interacts with FastAPI, which in turn queries the LLM via Ollama.
 
-The Streamlit UI (exposed on port 8501) communicates with the FastAPI backend (running internally on port 7860) to send user prompts and receive AI responses.
+---
 
-Get Started
-Simply type your query into the text box and click "Get Response" to interact with the AI assistant.
+## 🖥️ User Interface
 
-Built with Docker, Ollama, FastAPI, and Streamlit.
+By default, the **Streamlit UI** is the primary interface and launches at: