ydshieh commited on
Commit
f884ea7
1 Parent(s): 9eed0db
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -8,6 +8,16 @@ st.title("French Image Caption App")
8
  # For newline
9
  st.write('\n')
10
 
 
 
 
 
 
 
 
 
 
 
11
  #image = Image.open('samples/val_000000039769.jpg')
12
  #show = st.image(image, use_column_width=True)
13
  #show.image(image, 'Preloaded Image', use_column_width=True)
 
8
  # For newline
9
  st.write('\n')
10
 
11
+ st.markdown(
12
+ """
13
+ An image caption model by combining the ViT model and a French GPT2 model.\n
14
+ [Part of the [Huggingface JAX/Flax event](https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/).]\n
15
+ The pretained weights of both models are loaded, and a set of randomly initialized cross-attention weigths are added.\n
16
+ The model is trained on 65000 images from the COCO dataset for about 1500 steps.\n
17
+ The original english cpationis are translated to french for training purpose.\n
18
+ """
19
+ )
20
+
21
  #image = Image.open('samples/val_000000039769.jpg')
22
  #show = st.image(image, use_column_width=True)
23
  #show.image(image, 'Preloaded Image', use_column_width=True)