VDNT11 commited on
Commit
4012b48
1 Parent(s): a641f96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ if not os.path.exists('IndicTransToolkit'):
17
  from IndicTransToolkit import IndicProcessor
18
 
19
  # Initialize BLIP for image captioning
20
- blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
21
- blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base").to("cuda" if torch.cuda.is_available() else "cpu")
22
 
23
  # Function to generate captions
24
  def generate_caption(image_path):
 
17
  from IndicTransToolkit import IndicProcessor
18
 
19
  # Initialize BLIP for image captioning
20
+ blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
21
+ blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large").to("cuda" if torch.cuda.is_available() else "cpu")
22
 
23
  # Function to generate captions
24
  def generate_caption(image_path):