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