ccclllwww commited on
Commit
4204a24
·
verified ·
1 Parent(s): 243f6f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ import torch
18
  # Initialize image captioning pipeline with pretrained model
19
  _image_caption_pipeline = pipeline(
20
  task="image-to-text",
21
- model="noamrot/FuseCap_Image_Captioning"
22
  )
23
 
24
  # Global model configuration constants
 
18
  # Initialize image captioning pipeline with pretrained model
19
  _image_caption_pipeline = pipeline(
20
  task="image-to-text",
21
+ model="cnmoro/tiny-image-captioning"
22
  )
23
 
24
  # Global model configuration constants