rootsec1
/

gemma-2B-it-customer-support

Text Generation

text-generation-inference

Model card Files Files and versions

rootsec1 commited on Apr 15, 2024

Commit

d7c0e7e

·

verified ·

1 Parent(s): d2bb887

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -24,4 +24,5 @@ model = AutoModelForCausalLM.from_pretrained("your-username/your-model-name")
 chatbot = pipeline("text-generation", model=model, tokenizer=tokenizer)
 response = chatbot("How can I cancel my order?")
-print(response[0]['generated_text'])

 chatbot = pipeline("text-generation", model=model, tokenizer=tokenizer)
 response = chatbot("How can I cancel my order?")
+print(response[0]['generated_text'])
+```