added model selection option

app.py

@@ -4,7 +4,7 @@ from prefix_clip import download_pretrained_model, generate_caption
 from gpt2_story_gen import generate_story
 
 
-def main(pil_image, genre, model="Conceptual", use_beam_search=False):
+def main(pil_image, genre, model, use_beam_search=False):
     model_file = "pretrained_weights.pt"
 
     download_pretrained_model(model.lower(), file_to_save=model_file)
@@ -42,9 +42,10 @@ if __name__ == "__main__":
                     "sci_fi",
                 ],
             ),
+            gr.inputs.Radio(choices=["coco", "conceptual"], label="Model")
         ],
         outputs=gr.outputs.Textbox(label="Generated story"),
-        examples=[["car.jpg", "drama"], ['gangster.jpg', 'action']],
+        examples=[["car.jpg", "drama", "conceptual"], ["gangster.jpg", "action", "coco"]],
         enable_queue=True,
     )
     interface.launch()