Update README.md
Browse files
README.md
CHANGED
@@ -38,7 +38,12 @@ library_name: keras-hub
|
|
38 |
from keras_nlp.models import Gemma3CausalLM
|
39 |
from keras_nlp.samplers import TopKSampler
|
40 |
|
41 |
-
model = Gemma3CausalLM.from_preset("rudrashah/RLM_hingu")
|
42 |
-
|
|
|
|
|
|
|
|
|
|
|
43 |
print(output)
|
44 |
```
|
|
|
38 |
from keras_nlp.models import Gemma3CausalLM
|
39 |
from keras_nlp.samplers import TopKSampler
|
40 |
|
41 |
+
model = Gemma3CausalLM.from_preset("hf://rudrashah/RLM_hingu")
|
42 |
+
template = "Question:\n{question}\n\nAnswer:\n{answer}"
|
43 |
+
prompt = template.format(
|
44 |
+
question="Rudra acha ladka hai?",
|
45 |
+
answer="",
|
46 |
+
)
|
47 |
+
output = model.generate(prompt, max_length=256)
|
48 |
print(output)
|
49 |
```
|