gokaygokay commited on
Commit
2fd5ae5
1 Parent(s): 53bbd9c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -10,7 +10,7 @@ pipeline_tag: text2text-generation
10
  ---
11
 
12
  ```python
13
- from transformers import pipeline
14
 
15
  # Model checkpoint
16
  model_checkpoint = "gokaygokay/Flux-Prompt-Enhance"
 
10
  ---
11
 
12
  ```python
13
+ from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
14
 
15
  # Model checkpoint
16
  model_checkpoint = "gokaygokay/Flux-Prompt-Enhance"