Spaces:

Uzaiir
/

Image2Story

Sleeping

Uzaiir commited on May 27, 2025

Commit

5b23712

verified ·

1 Parent(s): 4718b5f

Update src/models/imageCaptioning.py

Files changed (1) hide show

src/models/imageCaptioning.py CHANGED Viewed

@@ -29,8 +29,8 @@ from transformers import BlipProcessor , BlipForConditionalGeneration
 import torch
-processor = BlipProcessor.from_pretrained("./models/Caption")
-model = BlipForConditionalGeneration.from_pretrained("./models/Caption")
 def generateCaption(image_path):

 import torch
+processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
+model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
 def generateCaption(image_path):