iamrobotbear commited on
Commit
3bf36bd
·
1 Parent(s): 58bb7b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -8
app.py CHANGED
@@ -10,14 +10,6 @@ model = Blip2ForConditionalGeneration.from_pretrained(
10
  "Salesforce/blip2-opt-2.7b", load_in_8bit=True, device_map='auto'
11
  )
12
 
13
- # Set device to GPU if available
14
- device = "cuda" if torch.cuda.is_available() else "cpu"
15
- model.to(device)
16
-
17
- # Set device to GPU if available
18
- device = "cuda" if torch.cuda.is_available() else "cpu"
19
- model.to(device)
20
-
21
  def blip2_interface(image, prompted_caption_text, vqa_question, chat_context):
22
  # Prepare image input
23
  image_input = Image.fromarray(image).convert('RGB')
 
10
  "Salesforce/blip2-opt-2.7b", load_in_8bit=True, device_map='auto'
11
  )
12
 
 
 
 
 
 
 
 
 
13
  def blip2_interface(image, prompted_caption_text, vqa_question, chat_context):
14
  # Prepare image input
15
  image_input = Image.fromarray(image).convert('RGB')