Update app.py
Browse files
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-
|
14 |
model = Blip2ForConditionalGeneration.from_pretrained(
|
15 |
-
"Salesforce/blip2-
|
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
|