Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
|
|
3 |
from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel
|
4 |
|
5 |
# Load text generation model
|
6 |
-
|
7 |
|
8 |
# Load image captioning model
|
9 |
encoder_checkpoint = "nlpconnect/vit-gpt2-image-captioning"
|
|
|
3 |
from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel
|
4 |
|
5 |
# Load text generation model
|
6 |
+
text_generation_model = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
|
7 |
|
8 |
# Load image captioning model
|
9 |
encoder_checkpoint = "nlpconnect/vit-gpt2-image-captioning"
|