Vijayendra
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -17,7 +17,7 @@ pipeline_tag: text-generation
|
|
17 |
import torch
|
18 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
19 |
|
20 |
-
#
|
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 |
-
#
|
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 |
-
#
|
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}")
|