Hugging Face's logo

Spaces:
flax-community
/
image-captioning
Runtime error

App Files Community
3
ydshieh commited on
Commit
a194253
•
1 Parent(s): e755009

fix syntax

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -9,7 +9,7 @@ from transformers import GPT2Tokenizer
9
  current_path = os.path.dirname(os.path.abspath(__file__))
10
  sys.path.append(current_path)
11
 
12
- Main model - ViTGPT2LM
13
  from vit_gpt2.modeling_flax_vit_gpt2_lm import FlaxViTGPT2LMForConditionalGeneration
14
 
15
  model_name_or_path = 'flax-community/vit-gpt2/checkpoints/ckpt_5/'
 
9
  current_path = os.path.dirname(os.path.abspath(__file__))
10
  sys.path.append(current_path)
11
 
12
+ # Main model - ViTGPT2LM
13
  from vit_gpt2.modeling_flax_vit_gpt2_lm import FlaxViTGPT2LMForConditionalGeneration
14
 
15
  model_name_or_path = 'flax-community/vit-gpt2/checkpoints/ckpt_5/'