noequal commited on
Commit
71a1ad1
·
1 Parent(s): bb3698f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,8 +25,8 @@ text_gen_tokenizer = GPT2Tokenizer.from_pretrained(model_name)
25
  text_gen_tokenizer.pad_token = text_gen_tokenizer.eos_token
26
 
27
  # Load your sequence classification model (e.g., BERT)
28
- seq_classifier_model = GPT2ForSequenceClassification.from_pretrained("fine_tuned_classifier_model")
29
- seq_classifier_tokenizer = GPT2Tokenizer.from_pretrained("fine_tuned_classifier_model")
30
  seq_classifier_tokenizer.pad_token = seq_classifier_tokenizer.eos_token
31
 
32
  # Create a title and a text input for the app
 
25
  text_gen_tokenizer.pad_token = text_gen_tokenizer.eos_token
26
 
27
  # Load your sequence classification model (e.g., BERT)
28
+ seq_classifier_model = GPT2ForSequenceClassification.from_pretrained(model_name)
29
+ seq_classifier_tokenizer = GPT2Tokenizer.from_pretrained(model_name)
30
  seq_classifier_tokenizer.pad_token = seq_classifier_tokenizer.eos_token
31
 
32
  # Create a title and a text input for the app