Update README.md

README.md

@@ -60,15 +60,15 @@ This model is not meant as a decision support system in the wild, only for educa
 
 ## How to Get Started with the Model
 
-Try it here : [Pseudolab/GaiaMiniMed](https://huggingface.co/spaces/pseudolab/GaiaMiniMed)
+- Try it here : [Pseudolab/GaiaMiniMed](https://huggingface.co/spaces/pseudolab/GaiaMiniMed)
 
-See the [author's demo](https://huggingface.co/spaces/tonic/gaiaminimed)
+- See the [author's demo](https://huggingface.co/spaces/tonic/gaiaminimed)
 
-Use the code below to get started with the model.
+- Use the code below to get started with the model.
 
 ```python
 
-# Gaia Mini Med Quick Start
+# Gaia MiniMed ⚕️🦅 Quick Start
 
 from transformers import AutoConfig, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, MistralForCausalLM
 from peft import PeftModel, PeftConfig
@@ -145,11 +145,11 @@ iface.launch()
 
 ```
 
-See the code below for more advanced deployment , including a naive memory store and user controllable parameters:
+- See the code below for more advanced deployment , including a naive memory store and user controllable parameters:
 
 ```Python
 
-# Welcome to a more advanced implementation of Gaia Mini Med - Falcon Chat
+# Gaia MiniMed⚕️🦅Falcon Chat
 
 from transformers import AutoConfig, AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel, PeftConfig
@@ -239,14 +239,20 @@ falcon_bot = FalconChatBot()
 title = "👋🏻Welcome to Tonic's 🦅Falcon's Medical👨🏻‍⚕️Expert Chat🚀"
 description = "You can use this Space to test out the GaiaMiniMed model [(Tonic/GaiaMiniMed)](https://huggingface.co/Tonic/GaiaMiniMed) or duplicate this Space and use it locally or on 🤗HuggingFace. [Join me on Discord to build together](https://discord.gg/VqTxc76K3u)."
 
-examples = [  
-    [  
-        "Assistant is a public health and medical expert named Gaia ready to help the user.",   
-        "Hi there, I have a question!",   
-        "My name is Gaia, I'm a health and sanitation expert ready to answer your medical questions.",   
-        "Assistant is a medical and sanitation question expert trained to answer medical questions", 0.4, 700, 0.90, 1.9  
+examples = [
+    [
+        {
+            "user_message": "Hi there, I have a question!",
+            "assistant_message": "My name is Gaia, I'm a health and sanitation expert ready to answer your medical questions.",
+            "history": [],
+            "temperature": 0.4,
+            "max_new_tokens": 700,
+            "top_p": 0.90,
+            "repetition_penalty": 1.9,
+        }
     ]
-]  
+]
+
 
 
 additional_inputs=[
@@ -295,9 +301,7 @@ iface = gr.Interface(
     description=description,
     examples=examples,
     inputs=[
-        gr.inputs.Textbox(label="System Prompt", type="text", lines=2),
-        gr.inputs.Textbox(label="User Message", type="text", lines=3),
-        gr.inputs.Textbox(label="Assistant Message", type="text", lines=2),
+        gr.inputs.Textbox(label="Input Parameters", type="text", lines=5),
     ] + additional_inputs,
     outputs="text",
     theme="ParityError/Anime"