wiusdy commited on
Commit
a48ebf4
1 Parent(s): fb006ce

add auto processor

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -1,9 +1,9 @@
1
- from transformers import BlipProcessor, BlipForQuestionAnswering
2
  from transformers.utils import logging
3
 
4
  class Inference:
5
  def __init__(self):
6
- self.blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-vqa-base")
7
  self.blip_model_saffal = BlipForQuestionAnswering.from_pretrained("wiusdy/blip_pretrained_saffal_fashion_finetuning")
8
  self.blip_model_control_net = BlipForQuestionAnswering.from_pretrained("wiusdy/blip_pretrained_control_net_fashion_finetuning")
9
  logging.set_verbosity_info()
 
1
+ from transformers import AutoProcessor, BlipForQuestionAnswering
2
  from transformers.utils import logging
3
 
4
  class Inference:
5
  def __init__(self):
6
+ self.blip_processor = AutoProcessor.from_pretrained("Salesforce/blip-vqa-base")
7
  self.blip_model_saffal = BlipForQuestionAnswering.from_pretrained("wiusdy/blip_pretrained_saffal_fashion_finetuning")
8
  self.blip_model_control_net = BlipForQuestionAnswering.from_pretrained("wiusdy/blip_pretrained_control_net_fashion_finetuning")
9
  logging.set_verbosity_info()