Update app.py

app.py

@@ -8,7 +8,7 @@ from PIL import Image
 from transformers import Blip2Processor, Blip2ForConditionalGeneration, InstructBlipProcessor, InstructBlipForConditionalGeneration
 
 
-def load_caption_model(blip2=False, instructblip=True):
+def load_caption_model(blip2=True, instructblip=False):
 
     if blip2:
         processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b", load_in_8bit=True,torch_dtype=torch.float16, device_map="cuda")
@@ -32,7 +32,7 @@ def answer_question(image, question, model, processor):
 
     inputs = processor(image, question, return_tensors="pt").to("cuda", torch.float16)
 
-    out = model.generate(**inputs, max_length=200, min_length=20).to("cuda", torch.float16)
+    out = model.generate(**inputs, max_length=200, min_length=20)
 
     answer = processor.decode(out[0], skip_special_tokens=True).strip()
     return answer