Update README.md

README.md

@@ -20,38 +20,21 @@ Model based on `bardsai/jaskier-7b-dpo-v5.6` (downstream version of Mistral7B) f
 
 You can use this model directly with a Hugging Face pipeline:
 ```python
-import ...
 
-base_model_name = "bardsai/jaskier-7b-dpo-v5.6"
-p = pipeline("conversational", model=base_model_name, torch_dtype=torch.float16, device_map="auto")
-messages = [{'messages':[{
-    "role": "user",
-    "content": "How much is 2+2"
-}]},
-{'messages':[{
-    "role": "user",
-    "content": "Tell me where the nearest water park is?"
-}]},
-]
+from transformers import pipeline, Conversation
+import torch
 
-def conversations():
-    for x in tqdm(messages):
-        yield Conversation(x['messages'])
-
-generate_kwargs = {
-    "do_sample": True,
-    "temperature": 0.7,
-    "max_new_tokens": 1024,
-}
-model_outputs = [out[-1]['content'] for out in p(conversations(), **generate_kwargs)]
-
-print(model_outputs)
+base_model_name = "bardsai/jaskier-7b-dpo-v6.1"
+chatbot = pipeline("conversational", model=base_model_name, torch_dtype=torch.float16, device_map="auto")
+conversation = Conversation("Can Poland into space?")
+conversation = chatbot(conversation)
+print(conversation.messages[-1]["content"])
 
 ```
 
 ## Output
 
->["The answer to the arithmetic operation of adding two to two is 4. In mathematical notation, it's written as 2 + 2 = 4.", "As an AI, I don't have physical proximity or real-time environment awareness. To provide location-based information, I need to be integrated with a specific location-tracking system or connected to a comprehensive database. Please check for local business listings, travel guides, or conduct an online search to find the nearest water park according to your current location."]
+"Poland, as a nation, doesn't physically travel to space. However, Poland has contributed to the field of space exploration through its scientists, engineers, and collaborations with international space agencies. The Polish Space Agency, established in 2016, aims to promote and coordinate the country's space activities."
 
 ## Changelog