Hugging Face's logo

Spaces:
flax-community
/
image-captioning
Runtime error

App Files Community
3
ydshieh commited on
Commit
e79fa8b
·
1 Parent(s): 340afca

fix model path

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -12,7 +12,7 @@ sys.path.append(current_path)
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/'
16
  flax_vit_gpt2_lm = FlaxViTGPT2LMForConditionalGeneration.from_pretrained(model_name_or_path)
17
 
18
  def predict(image):
 
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'
16
  flax_vit_gpt2_lm = FlaxViTGPT2LMForConditionalGeneration.from_pretrained(model_name_or_path)
17
 
18
  def predict(image):