Spaces:
Sleeping
Sleeping
add auto processor
Browse files- inference.py +2 -2
inference.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
-
from transformers import
|
2 |
from transformers.utils import logging
|
3 |
|
4 |
class Inference:
|
5 |
def __init__(self):
|
6 |
-
self.blip_processor =
|
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()
|