Update README.md

README.md

@@ -1,63 +1,67 @@
-# 🫐 BerryAI
-
-BerryAI is a multimodal, coding-focused chatbot trained on a mixture of conversational, coding, UI design, and math reasoning datasets.  
-It blends creative dialogue with factual accuracy and programming skill.
-
 ---
-
-## 🚀 Features
-- 💬 Conversational: Fine-tuned with UltraChat for engaging responses  
-- 🧠 Smart: FineWeb improves general knowledge  
-- 🧮 Reasoning: OpenWebMath for step-by-step logic  
-- 💻 Coder: WizardCoderData, StarCoderData, and DS-Coder-Instruct for deeper programming skill  
-- 🎨 UI Genius: Trained on multiple UI/UX datasets for design insights
-
+language: 
+- en
+license: mit
+tags:
+- coding
+- ui
+- chat
+- math
+- factual
+- agent
+- multimodal
+base_model:
+- mistralai/Mixtral-8x7B-v0.1
+- OpenBuddy/openbuddy-openllama-7b-v12-bf16
+- HuggingFaceH4/mistral-7b-grok
+- togethercomputer/RedPajama-INCITE-7B-Chat
+datasets:
+- hotboxxgenn/mix-openhermes-openorca-platypus-airoboros-chatalpaca-opencode
+- microsoft/rStar-Coder
+- ed001/ds-coder-instruct-v1
+- bigcode/starcoderdata
+- bigcode/starcoder2data-extras
+- codeparrot/self-instruct-starcoder
+- mrtoy/mobile-ui-design
+- YashJain/UI-Elements-Detection-Dataset
+- tecky-tech/Tecky-UI-Elements-VLM
+- Tesslate/UIGEN-T2
+- FineWeb
+- OpenWebMath
+- UltraChat
+- WizardCoderData
+library_name: transformers
 ---
 
-## 🧰 Base Models
-- `mistralai/Mixtral-8x7B-v0.1`  
-- `OpenBuddy/openbuddy-openllama-7b-v12-bf16`  
-- `HuggingFaceH4/mistral-7b-grok`  
-- `togethercomputer/RedPajama-INCITE-7B-Chat`
+# 🍓 BerryAI
 
----
-
-## 📚 Datasets
-- hotboxxgenn/mix-openhermes-openorca-platypus-airoboros-chatalpaca-opencode  
-- microsoft/rStar-Coder  
-- ed001/ds-coder-instruct-v1  
-- bigcode/starcoderdata  
-- bigcode/starcoder2data-extras  
-- codeparrot/self-instruct-starcoder  
-- mrtoy/mobile-ui-design  
-- YashJain/UI-Elements-Detection-Dataset  
-- tecky-tech/Tecky-UI-Elements-VLM  
-- Tesslate/UIGEN-T2  
-- FineWeb  
-- OpenWebMath  
-- UltraChat  
-- WizardCoderData  
+**Author:** [@hotboxxgenn](https://huggingface.co/hotboxxgenn)  
+**Version:** 1.1  
+**Type:** Conversational + Coding + UI + Math + Factual Model  
+**Base:** Mixtral 8x7B, OpenBuddy, Mistral-Grok, RedPajama
 
 ---
 
-## 🧠 Capabilities
-✅ Coding (Python, C#, JavaScript)  
-✅ UI Design Advice (Mobile / Web)  
-✅ Conversational + Creative Writing  
-✅ Factual Knowledge  
-✅ Math & Logic Reasoning
+## ✨ Overview
+
+BerryAI is a **multi-skill LLM** designed to perform:
+- 💻 **Coding** — Python, JS, React, Tailwind, multi-step reasoning  
+- 🎨 **UI generation** — generate clean, responsive interfaces  
+- 💬 **Conversational chat** — helpful, creative, engaging tone  
+- 🧮 **Math reasoning** — step-by-step calculations  
+- 🔍 **Factual grounding** — reduced hallucination, improved accuracy  
 
 ---
 
-## ⚡ Usage
+## 🚀 Usage
+
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_id = "hotboxxgenn/BerryAI"
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id)
+tokenizer = AutoTokenizer.from_pretrained("hotboxxgenn/BerryAI")
+model = AutoModelForCausalLM.from_pretrained("hotboxxgenn/BerryAI")
 
-prompt = "Write a C# function that scrapes titles from a webpage."
+prompt = "Generate a responsive React login form with Tailwind CSS."
 inputs = tokenizer(prompt, return_tensors="pt")
-outputs = model.generate(**inputs, max_new_tokens=250)
+outputs = model.generate(**inputs, max_new_tokens=200)
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))