1inkusFace commited on
Commit
7d3c40d
·
verified ·
1 Parent(s): d5f0fef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -159,6 +159,7 @@ def load_and_prepare_model():
159
  pipe = load_and_prepare_model()
160
 
161
  # text models
 
162
  captioner = pipeline(model="ydshieh/vit-gpt2-coco-en",device='cuda', task="image-to-text")
163
  captioner_2 = pipeline(model="Salesforce/blip-image-captioning-base",device='cuda', task="image-to-text")
164
  captioner_3 = pipeline(model="Salesforce/blip-image-captioning-large",device='cuda', task="image-to-text")
 
159
  pipe = load_and_prepare_model()
160
 
161
  # text models
162
+ checkpoint = "microsoft/Phi-3.5-mini-instruct"
163
  captioner = pipeline(model="ydshieh/vit-gpt2-coco-en",device='cuda', task="image-to-text")
164
  captioner_2 = pipeline(model="Salesforce/blip-image-captioning-base",device='cuda', task="image-to-text")
165
  captioner_3 = pipeline(model="Salesforce/blip-image-captioning-large",device='cuda', task="image-to-text")