Hugging Face's logo

sartajbhuvaji
/
gutenberg-gpt2

Text Classification
Transformers
Safetensors
English
gpt2
text-generation-inference
Inference Endpoints
Model card Files Community
sartajbhuvaji commited on
Commit
f1c510e
·
verified ·
1 Parent(s): 671845b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -19,7 +19,7 @@ from transformers import pipeline
19
  import pandas as pd
20
 
21
  # Load the model from Hugging Face
22
- model = GPT2ForSequenceClassification.from_pretrained('sartajbhuvaji/gutenberg-gpt2', num_labels=num_labels)
23
  tokenizer = GPT2Tokenizer.from_pretrained("sartajbhuvaji/gutenberg-gpt2")
24
 
25
  # Create a text classification pipeline
 
19
  import pandas as pd
20
 
21
  # Load the model from Hugging Face
22
+ model = GPT2ForSequenceClassification.from_pretrained('sartajbhuvaji/gutenberg-gpt2')
23
  tokenizer = GPT2Tokenizer.from_pretrained("sartajbhuvaji/gutenberg-gpt2")
24
 
25
  # Create a text classification pipeline