Update README.md

README.md

@@ -95,7 +95,7 @@ quantization_config = BitsAndBytesConfig(
     bnb_4bit_use_double_quant=True,
 )
 
-model_id = "EpistemeAI/Fireball-Meta-Llama-3.1-8B-Instruct-Agent-0.005-128K-code-COT"
+model_id = EpistemeAI2/Fireball-Meta-Llama-3.1-8B-Instruct-Agent-0.005-128K-code-COT"
 pipeline = transformers.pipeline(
     "text-generation",
     model=model_id,
@@ -104,12 +104,13 @@ pipeline = transformers.pipeline(
 )
 messages = [
     {"role": "system", "content":  """
-    You should reason about the input and provide a logical explanation.
-The explanation should follow these rules:
-- The explanation should be written at graduate level engineering, science, math and literature
-- The explanation should be split into subtasks
-- The explanation should always end with 2-3 related concepts.
-- subtasks have their own chain of thoughts
+    Environment: ipython. Tools: brave_search, wolfram_alpha. Cutting Knowledge Date: December 2023. Today Date: 4 October 2024\n
+    You are a coding assistant with expert with everything\n
+    Ensure any code you provide can be executed \n
+    with all required imports and variables defined. List the imports.  Structure your answer with a description of the code solution. \n
+    write only the code. do not print anything else.\n
+    debug code if error occurs. \n
+    Here is the user question: {question}
     """},
     {"role": "user", "content": "Create a bar plot showing the market capitalization of the top 7 publicly listed companies using matplotlib"}
 ]
@@ -141,7 +142,7 @@ fourbit_models = [
 ] # More models at https://huggingface.co/unsloth
 
 model, tokenizer = FastLanguageModel.from_pretrained(
-    model_name = "EpistemeAI/Fireball-Meta-Llama-3.1-8B-Instruct-Agent-0.005-128K-code-COT",
+    model_name = "EpistemeAI2/Fireball-Meta-Llama-3.1-8B-Instruct-Agent-0.005-128K-code-COT",
     max_seq_length = 128000,
     load_in_4bit = True,
     token =userdata.get('HF_TOKEN')
@@ -150,12 +151,14 @@ def chatbot(query):
   messages = [
       {"from": "system", "value":
        """
-       You should reason about the input and provide a logical explanation.
-The explanation should follow these rules:
-- The explanation should be written at graduate level engineering, science, math and literature
-- The explanation should be split into subtasks
-- The explanation should always end with 2-3 related concepts.
-- subtasks have their own chain of thoughts
+      Environment: ipython. Tools: brave_search, wolfram_alpha. Cutting Knowledge Date: December 2023. Today Date: 4 October 2024\n
+      You are a coding assistant with expert with everything\n
+      Ensure any code you provide can be executed \n
+      with all required imports and variables defined. List the imports.  Structure your answer with a description of the code solution. \n
+      write only the code. do not print anything else.\n
+      use ipython for search tool. \n
+      debug code if error occurs. \n
+      Here is the user question: {question}
       """
        },
       {"from": "human", "value": query},
@@ -166,6 +169,7 @@ The explanation should follow these rules:
   _ = model.generate(input_ids = inputs, streamer = text_streamer, max_new_tokens = 2048, use_cache = True)
 ```
 
+
 # Response
 
 ```python