Update README.md

README.md

@@ -12,27 +12,21 @@ Bloom (2.5 B) Scientific Model fine-tuned on Zen knowledge
    #####
 
 
+```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-
-tokenizer = AutoTokenizer.from_pretrained("MultiTrickFox/bloom-2b5_Zen")
+tokenizer = AutoTokenizer.from_pretrained("MultiTrickFox/bloom-2b5_Zen") 
 model = AutoModelForCausalLM.from_pretrained("MultiTrickFox/bloom-2b5_Zen")
 
-model.cuda()
+model
 tokenizer.pad_token_id = tokenizer.eos_token_id
 
-
 generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
 
+inp = [ """Today""", """Yesterday""" ]
 
-inp = [
-"""Today""",
-"""Yesterday"""
-]
-
-out = generator(
-    inp.cuda(),
-    do_sample=True,
+out = generator( 
+    inp, do_sample=True,
 
     temperature=.6,
     typical_p=.7,
@@ -43,5 +37,5 @@ out = generator(
     max_time=60, # seconds
 )
 
-
-for o in out: print(o[0]['generated_text'])
+for o in out: print(o[0]['generated_text'])
+```