Hugging Face's logo

Spaces:
HF-Pawan
/
AI-Image-Describer
Running

App Files Community
HF-Pawan commited on
Commit
658ebc4
·
1 Parent(s): f4a1ca1

Latest Changes

Browse files
Files changed (1) hide show
  1. core/model_loader.py +1 -1
core/model_loader.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
  from transformers import BlipProcessor, BlipForConditionalGeneration
3
 
4
- MODEL_ID = "Salesforce/blip-image-captioning-base"
5
  DEVICE = torch.device("cpu")
6
 
7
  # Prompt templates
 
1
  import torch
2
  from transformers import BlipProcessor, BlipForConditionalGeneration
3
 
4
+ MODEL_ID = "Salesforce/blip-image-captioning-large"
5
  DEVICE = torch.device("cpu")
6
 
7
  # Prompt templates