Update README.md

README.md

@@ -14,27 +14,29 @@ This model was trained using AutoTrain. For more information, please visit [Auto
 # Usage
 
 ```python
+!pip install transformers
 
-from transformers import AutoModelForCausalLM, AutoTokenizer
+!pip install accelerate
 
-model_path = "PATH_TO_THIS_REPO"
 
-tokenizer = AutoTokenizer.from_pretrained(model_path)
-model = AutoModelForCausalLM.from_pretrained(
-    model_path,
-    device_map="auto",
-    torch_dtype='auto'
-).eval()
+from huggingface_hub import notebook_login
+notebook_login()
 
-# Prompt content: "hi"
-messages = [
-    {"role": "user", "content": "hi"}
-]
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
+
+
+tokenizer = AutoTokenizer.from_pretrained("mjmanashti/autotrain-cff1t-gk81o")
+torch.set_default_dtype(torch.float16)
 
-input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')
-output_ids = model.generate(input_ids.to('cuda'))
-response = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
+model = AutoModelForCausalLM.from_pretrained("mjmanashti/autotrain-cff1t-gk81o", device_map="auto")
+
+chat = [
+    { "role": "user", "content": "Based on the following input data: [Time: 2024-01-29 23:00:00, Open: 1.0834, High: 1.0837, Low: 1.08334, Close: 1.08338, Volume: 722] what trading signal (BUY, SELL, or HOLD) should be executed to maximize profit? If the signal is BUY, what would be the entry price and If the signal is SELL, what would be the exit price for profit maximization? " },
+]
+prompt = tokenizer.apply_chat_template(chat, tokenize=False, add_generation_prompt=True)
+inputs = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt")
+outputs = model.generate(input_ids=inputs.to(model.device), max_new_tokens=150)
+print(tokenizer.decode(outputs[0]))
 
-# Model response: "Hello! How can I assist you today?"
-print(response)
 ```