Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
163 |
-
|
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")
|