mrSoul7766
/

CUSsupport-chat-t5-small

text2text-generation

text-generation-inference

customer support

Model card Files Files and versions

mrSoul7766 commited on Jan 17, 2024

Commit

51cd6a3

·

verified ·

1 Parent(s): 2bad44b

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -52,8 +52,8 @@ I'm on it! I'm here to assist you in tracking the compensation. To track the com
 # Load model directly
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-tokenizer = AutoTokenizer.from_pretrained("mrSoul7766/chat-support-t5-small")
-model = AutoModelForSeq2SeqLM.from_pretrained("mrSoul7766/chat-support-t5-small")
 # Set maximum generation length
 max_length = 512

 # Load model directly
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+tokenizer = AutoTokenizer.from_pretrained("mrSoul7766/CUSsupport-chat-t5-small")
+model = AutoModelForSeq2SeqLM.from_pretrained("mrSoul7766/CUSsupport-chat-t5-small")
 # Set maximum generation length
 max_length = 512