Add sidebar

app.py

@@ -12,6 +12,15 @@ tokenizer = AutoTokenizer.from_pretrained(CHECKPOINT)
 
 model.eval()
 
+st.title("Image Captioning with ViT & GePpeTto 🇮🇹")
+
+st.sidebar.markdown("## Generation parameters")
+max_length = st.sidebar.number_input("Max length", value=20, min_value=1)
+num_beams = st.sidebar.number_input("Beam size", value=5, min_value=1)
+early_stopping = st.sidebar.checkbox("Early stopping", value=True)
+no_repeat_ngram_size= st.sidebar.number_input("no repeat ngrams size", value=2, min_value=1)
+num_return_sequences = st.sidebar.number_input("Generated sequences", value=3, min_value=1)
+
 
 def generate_caption(url):
     image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
@@ -31,8 +40,6 @@ def generate_caption(url):
     return captions[0]
 
 
-st.title("Captioning demo")
-
 url = st.text_input(
     "Insert your URL", "https://iheartcats.com/wp-content/uploads/2015/08/c84.jpg"
 )