1inkusFace commited on
Commit
904669d
·
verified ·
1 Parent(s): 79f5a4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -159,8 +159,8 @@ pipe = load_and_prepare_model()
159
 
160
  # text models
161
  checkpoint = "microsoft/Phi-3.5-mini-instruct"
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="ford442/blip-image-to-text-large-bf16",device='cuda', task="image-to-text")
165
  model5 = Blip2ForConditionalGeneration.from_pretrained("ford442/blip2-image-to-text-bf16").to('cuda')
166
  processor5 = Blip2Processor.from_pretrained("ford442/blip2-image-to-text-bf16")
 
159
 
160
  # text models
161
  checkpoint = "microsoft/Phi-3.5-mini-instruct"
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="ford442/blip-image-to-text-large-bf16",device='cuda', task="image-to-text")
165
  model5 = Blip2ForConditionalGeneration.from_pretrained("ford442/blip2-image-to-text-bf16").to('cuda')
166
  processor5 = Blip2Processor.from_pretrained("ford442/blip2-image-to-text-bf16")