ManishThota commited on
Commit
2139b27
·
verified ·
1 Parent(s): 7e6ce8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ from transformers import BlipProcessor, BlipForQuestionAnswering
5
 
6
  # Initialize the model and processor
7
  processor = BlipProcessor.from_pretrained("Salesforce/blip-vqa-base")
8
- # model = BlipForQuestionAnswering.from_pretrained("ManishThota/InstructBlip-VQA")
9
- model = BlipForQuestionAnswering.from_pretrained("Salesforce/blip-vqa-base")
10
 
11
  def predict_answer(image, question):
12
  # Convert PIL image to RGB if not already
 
5
 
6
  # Initialize the model and processor
7
  processor = BlipProcessor.from_pretrained("Salesforce/blip-vqa-base")
8
+ model = BlipForQuestionAnswering.from_pretrained("ManishThota/InstructBlip-VQA")
9
+ # model = BlipForQuestionAnswering.from_pretrained("Salesforce/blip-vqa-base")
10
 
11
  def predict_answer(image, question):
12
  # Convert PIL image to RGB if not already