Hugging Face's logo

Spaces:
m7mdal7aj
/
KB-VQA
Sleeping

App Files Community
2
m7mdal7aj commited on
Commit
d5a60de
•
1 Parent(s): 9d0134c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import accelerate
5
  import scipy
6
  from PIL import Image
7
 
8
- from transformers import Blip2Processor, Blip2ForConditionalGeneration
9
 
10
 
11
  def load_caption_model(blip2=False, instructblip=True):
 
5
  import scipy
6
  from PIL import Image
7
 
8
+ from transformers import Blip2Processor, Blip2ForConditionalGeneration, InstructBlipProcessor, InstructBlipForConditionalGeneration
9
 
10
 
11
  def load_caption_model(blip2=False, instructblip=True):