Jyothirmai commited on
Commit
9cda1d3
1 Parent(s): db9f8b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ with gr.Row():
45
  max_tokens = gr.Dropdown(list(range(50, 101)), label="Max Tokens", value=75)
46
  temperature = gr.Slider(0.5, 0.9, step=0.1, label="Temperature", value=0.7)
47
 
48
- model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model")
49
  generate_button = gr.Button("Generate Caption")
50
 
51
 
 
45
  max_tokens = gr.Dropdown(list(range(50, 101)), label="Max Tokens", value=75)
46
  temperature = gr.Slider(0.5, 0.9, step=0.1, label="Temperature", value=0.7)
47
 
48
+ model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention", "Baseline Model CNN-RNN"], label="Select Model")
49
  generate_button = gr.Button("Generate Caption")
50
 
51