thorfine commited on
Commit
2321f72
·
verified ·
1 Parent(s): 554a7f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,9 +10,9 @@ import whisper
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
 
12
  # Load BLIP-2 model
13
- processor = Blip2Processor.from_pretrained("Salesforce/blip2-flan-t5-xl")
14
  model = Blip2ForConditionalGeneration.from_pretrained(
15
- "Salesforce/blip2-flan-t5-xl", device_map="auto"
16
  ).to(device)
17
 
18
  # Load Whisper model
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
 
12
  # Load BLIP-2 model
13
+ processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b")
14
  model = Blip2ForConditionalGeneration.from_pretrained(
15
+ "Salesforce/blip2-opt-2.7b", device_map="auto"
16
  ).to(device)
17
 
18
  # Load Whisper model