Fixed tabs

app.py

@@ -5,14 +5,10 @@ from zero_shot_classification import get_zero_shot_classification_tab
 
 
 with gr.Blocks() as app:
-    image_captioning_tab = get_image_captioning_tab()
-    visual_qa_tab = get_visual_qa_tab()
-    zero_shot_classification_tab = get_zero_shot_classification_tab()
-
-    gr.TabbedInterface(
-        [image_captioning_tab, visual_qa_tab, zero_shot_classification_tab], 
-        ["Image Captioning", "Visual Q&A", "Zero-Shot Classification"]
-    )
+    with gr.Tabs():
+        get_image_captioning_tab()
+        get_visual_qa_tab()
+        get_zero_shot_classification_tab()
 
 
 app.launch()

image_captioning.py

@@ -28,7 +28,7 @@ def get_image_captioning_tab():
 
         return [result, time_spent]
     
-    with gr.Blocks() as image_captioning_tab:
+    with gr.TabItem("Image Captioning") as image_captioning_tab:
         gr.Markdown("# Image Captioning")
 
         with gr.Row():

visual_qa.py

@@ -25,7 +25,7 @@ def get_visual_qa_tab():
 
         return [result, time_spent]
     
-    with gr.Blocks() as visual_qa_tab:
+    with gr.TabItem("Visual Q&A") as visual_qa_tab:
         gr.Markdown("# Visual Question & Answering")
 
         with gr.Row():

zero_shot_classification.py

@@ -35,7 +35,7 @@ def get_zero_shot_classification_tab():
 
         return [result, time_spent]
     
-    with gr.Blocks() as zero_shot_image_classification_tab:
+    with gr.TabItem("Zero-Shot Classification") as zero_shot_image_classification_tab:
         gr.Markdown("# Zero-Shot Image Classification")
 
         with gr.Row():