dwb2023 commited on
Commit
68fd1d0
·
verified ·
1 Parent(s): b640a79

update Imports for Transformers library

Browse files
Files changed (1) hide show
  1. app_dialogue.py +1 -1
app_dialogue.py CHANGED
@@ -22,7 +22,7 @@ import io
22
  import datasets
23
 
24
  import gradio as gr
25
- from transformers import AutoImageProcessor, TextIteratorStreamer
26
  from transformers import AutoModelForVision2Seq
27
 
28
 
 
22
  import datasets
23
 
24
  import gradio as gr
25
+ from transformers import AutoImageProcessor, AutoTokenizer, TextIteratorStreamer, StoppingCriteria
26
  from transformers import AutoModelForVision2Seq
27
 
28