Spaces:
Runtime error
Runtime error
File size: 447 Bytes
686f21e 8d9306e 686f21e 8d9306e 686f21e 8d9306e 686f21e 8d9306e 686f21e 8d9306e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import os, sys
import numpy as np
from PIL import Image
import jax
from transformers import ViTFeatureExtractor
from transformers import GPT2Tokenizer
current_path = os.path.dirname(os.path.abspath(__file__))
sys.path.append(current_path)
# Main model - ViTGPT2LM
# from vit_gpt2.modeling_flax_vit_gpt2_lm import FlaxViTGPT2LMForConditionalGeneration
def predict(image):
return 'dummy caption!', ['dummy', 'caption', '!'], [1, 2, 3]
|