Vijayendra commited on
Commit
3c3b33d
·
verified ·
1 Parent(s): c36136e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -17,7 +17,7 @@ pipeline_tag: text-generation
17
  import torch
18
  from transformers import AutoModelForCausalLM, AutoTokenizer
19
 
20
- # 🏆 **Model Name on Hugging Face**
21
  MODEL_NAME = "Vijayendra/DeepSeek-Llama3.1-8B-DeepThinker-v1"
22
 
23
  # 🛠 **Load Model & Tokenizer from Hugging Face**
@@ -52,14 +52,14 @@ def generate_response(model, tokenizer, prompt, max_new_tokens=2048, temperature
52
  # Decode response
53
  return tokenizer.decode(generated_tokens[0], skip_special_tokens=True)
54
 
55
- # 🏆 **Test Questions**
56
  questions = [
57
  "What happened yesterday?",
58
  "If an unstoppable force hits an immovable object, what happens?",
59
  "The sun orbits the Earth once every 365 days. Is this true?"
60
  ]
61
 
62
- # 🏆 **Generate and Print Responses**
63
  for i, question in enumerate(questions, 1):
64
  response = generate_response(model, tokenizer, question)
65
  print(f"\n🟢 Question {i}: {question}")
 
17
  import torch
18
  from transformers import AutoModelForCausalLM, AutoTokenizer
19
 
20
+ # **Model Name on Hugging Face**
21
  MODEL_NAME = "Vijayendra/DeepSeek-Llama3.1-8B-DeepThinker-v1"
22
 
23
  # 🛠 **Load Model & Tokenizer from Hugging Face**
 
52
  # Decode response
53
  return tokenizer.decode(generated_tokens[0], skip_special_tokens=True)
54
 
55
+ # **Test Questions**
56
  questions = [
57
  "What happened yesterday?",
58
  "If an unstoppable force hits an immovable object, what happens?",
59
  "The sun orbits the Earth once every 365 days. Is this true?"
60
  ]
61
 
62
+ # **Generate and Print Responses**
63
  for i, question in enumerate(questions, 1):
64
  response = generate_response(model, tokenizer, question)
65
  print(f"\n🟢 Question {i}: {question}")