Update README.md

README.md

@@ -20,6 +20,7 @@ pipeline_tag: text-generation
 
 ## Model Details
 
+[![Open in Google Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1WYhhfvFzQukGzEqWHu3gKmigStJTjWxV?usp=sharing)
 - Model: [nferroukhi/WizardLM-Uncensored-Falcon-7b-sharded-bf16](https://huggingface.co/nferroukhi/WizardLM-Uncensored-Falcon-7b-sharded-bf16)
 - Base Model: [ehartford/WizardLM-Uncensored-Falcon-7b](https://huggingface.co/ehartford/WizardLM-Uncensored-Falcon-7b)
 - Fine-tuned with: [MBZUAI/Bactrian-X (Indonesian subset)](https://huggingface.co/datasets/MBZUAI/Bactrian-X/viewer/id/train)
@@ -38,7 +39,6 @@ pip install -U bitsandbytes==0.39.0
 pip install -U einops==0.6.1
 ```
 
-
 ## How to Use
 
 ### Stream Output
@@ -66,12 +66,24 @@ model = PeftModel.from_pretrained(model, "azale-ai/DukunLM-Uncensored-7B")
 tokenizer = AutoTokenizer.from_pretrained("azale-ai/DukunLM-Uncensored-7B")
 streamer = TextStreamer(tokenizer)
 
-input_prompt = "Jelaskan mengapa air penting bagi kehidupan manusia."
+instruction_prompt = "Jelaskan mengapa air penting bagi kehidupan manusia."
+input_prompt = ""
 
-text = f"""
-Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
+if input_prompt == "":
+  text = f"""Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
 
 ### Instruction:
+{instruction_prompt}
+
+### Response:
+"""
+else:
+    text = f"""Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
+
+### Instruction:
+{instruction_prompt}
+
+### Input:
 {input_prompt}
 
 ### Response:
@@ -113,12 +125,24 @@ model = AutoModelForCausalLM.from_pretrained(
 model = PeftModel.from_pretrained(model, "azale-ai/DukunLM-Uncensored-7B")
 tokenizer = AutoTokenizer.from_pretrained("azale-ai/DukunLM-Uncensored-7B")
 
-input_prompt = "Jelaskan mengapa air penting bagi kehidupan manusia."
+instruction_prompt = "Bangun dialog chatbot untuk layanan pelanggan yang ingin membantu pelanggan memesan produk tertentu."
+input_prompt = "Produk: Sepatu Nike Air Max"
 
-text = f"""
-Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
+if input_prompt == "":
+  text = f"""Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
 
 ### Instruction:
+{instruction_prompt}
+
+### Response:
+"""
+else:
+    text = f"""Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
+
+### Instruction:
+{instruction_prompt}
+
+### Input:
 {input_prompt}
 
 ### Response: