Hugging Face's logo

Spaces:
dwb2023
/
idefics2_playground
Paused

App Files Community
dwb2023 commited on
Commit
9bb961c
·
verified ·
1 Parent(s): 9f775ec

Update app_dialogue.py

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 AutoModelForVision2Seq, AutoTokenizer, AutoImageProcessor, LogitsProcessor
26
 
27
  # model_name_or_path = "Salesforce/xgen-mm-phi3-mini-instruct-interleave-r-v1.5"
28
  # model = AutoModelForVision2Seq.from_pretrained(model_name_or_path, trust_remote_code=True)
 
22
  import datasets
23
 
24
  import gradio as gr
25
+ from transformers import AutoModelForVision2Seq, AutoTokenizer, AutoImageProcessor, LogitsProcessor, TextIteratorStreamer
26
 
27
  # model_name_or_path = "Salesforce/xgen-mm-phi3-mini-instruct-interleave-r-v1.5"
28
  # model = AutoModelForVision2Seq.from_pretrained(model_name_or_path, trust_remote_code=True)