Update app.py

app.py

@@ -3,18 +3,13 @@ from transformers import AutoProcessor, Blip2ForConditionalGeneration
 import torch
 from PIL import Image
 
+# Set device to GPU if available
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
 # Load the BLIP-2 model and processor
-# Load the BLIP-2 model and processor without int8 quantization
 processor = AutoProcessor.from_pretrained("Salesforce/blip2-opt-2.7b")
 model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b").to(device)
 
-
-
-
-
-# Set device to GPU if available
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
 def blip2_interface(image, prompted_caption_text, vqa_question, chat_context):
     # Prepare image input
     image_input = Image.fromarray(image).convert('RGB')