Update README.md

README.md

@@ -25,10 +25,10 @@ This is a medical fine tuned model from the [Falcon-7b-Instruction](https://hugg
 - **License:** MIT
 - **Finetuned from model:**[tiiuae/falcon-7b-instruct](https://huggingface.co/tiiuae/falcon-7b-instruct)
 - 
-### Model Sources [optional]
+### Model Sources 
 
 - **Repository:** [Github](https://github.com/Josephrp/AI-challenge-hackathon/blob/master/falcon_7b_instruct_GaiaMiniMed_dataset.ipynb)
-- **Demo [optional]:** {{ demo | default("[More Information Needed]", true)}}
+- **Demo :** [pseudolab/gaiaminimed](https://huggingface.co/spaces/pseudolab/gaiaminimed) & [tonic/gaiaminimed](https://huggingface.com/spaces/tonic/gaiaminimed)
 
 ## Uses
 
@@ -81,8 +81,14 @@ tokenizer = AutoTokenizer.from_pretrained(base_model_id, trust_remote_code=True,
 tokenizer.pad_token = tokenizer.eos_token
 tokenizer.padding_side = 'left'
 
-# Load the Falcon model with the specified configuration
-falcon_model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-7b-instruct")
+
+# Load the GaiaMiniMed model with the specified configuration
+# Load the Peft model with a specific configuration
+peft_config = PeftConfig.from_pretrained("Tonic/GaiaMiniMed")
+peft_model = PeftModel.from_pretrained("Tonic/GaiaMiniMed", config=peft_config)
+
+# Now you can use peft_model without any NameError
+peft_model = peft_model.to_bettertransformer("tiiuae/falcon-7b-instruct")
 
 # Class to encapsulate the Falcon chatbot
 class FalconChatBot:
@@ -113,7 +119,7 @@ class FalconChatBot:
         for message in processed_history:
             user_message = message["user"]
             assistant_message = message["assistant"]
-            conversation += f"User: {user_message}\nFalcon:{' ' + assistant_message if assistant_message else ''}\n"
+            conversation += f"Falcon:{' ' + assistant_message if assistant_message else ''} User: {user_message}\n Falcon:\n"
 
         # Encode the formatted conversation using the tokenizer
         input_ids = tokenizer.encode(conversation, return_tensors="pt", add_special_tokens=False)
@@ -130,8 +136,8 @@ class FalconChatBot:
 falcon_bot = FalconChatBot()
 
 # Define the Gradio interface
-title = "👋🏻Welcome to Falcon's Medical Expert Chat🚀"
-description = "You can use this Space to test out the Falcon model [(tiiuae/falcon-7b-instruct)](https://huggingface.co/tiiuae/falcon-7b-instruct) or duplicate this Space and use it locally or on 🤗HuggingFace. [Join me on Discord to build together](https://discord.gg/VqTxc76K3u)."
+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 = [{
     "preprompt": "system message",
     "history": [{
@@ -149,7 +155,7 @@ iface = gr.Interface(
     description=description,
     examples=examples,
     inputs=[
-        gr.inputs.Textbox(label="Input Data", type="json),
+        gr.inputs.Textbox(label="Input Data", type="json"),
     ],
     outputs="text",
     theme="ParityError/Anime"