Jyothirmai
commited on
Update app.py
Browse files
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 |
|