aasd

app.py

@@ -73,22 +73,23 @@ def run_merge_and_push():
     return "Model merged and pushed to Hugging Face Hub successfully!"
 
 # Gradio interface
-iface = gr.Interface(
-    fn=generate_text,
-    inputs=gr.Textbox(lines=5, label="Input Text"),
-    outputs=gr.Textbox(lines=10, label="Generated Text"),
-    title="Merged Llama 3 8B Instruct Model",
-    description="Enter text to generate a response from the merged model.",
-)
-
-merge_and_push_button = gr.Button("Merge Models and Push to Hub")
-merge_and_push_output = gr.Textbox(lines=20, label="Merge and Push Output")
-
-merge_and_push_button.click(run_merge_and_push, outputs=merge_and_push_output)
+with gr.Blocks() as demo:
+    with gr.TabbedInterface(["Generate Text", "Merge and Push"]) as tabs:
+        with tabs[0]:
+            iface = gr.Interface(
+                fn=generate_text,
+                inputs=gr.Textbox(lines=5, label="Input Text"),
+                outputs=gr.Textbox(lines=10, label="Generated Text"),
+                title="Merged Llama 3 8B Instruct Model",
+                description="Enter text to generate a response from the merged model.",
+            )
+            iface.launch()
+        
+        with tabs[1]:
+            merge_and_push_button = gr.Button("Merge Models and Push to Hub")
+            merge_and_push_output = gr.Textbox(lines=20, label="Merge and Push Output")
+            merge_and_push_button.click(run_merge_and_push, outputs=merge_and_push_output)
 
 # Launch the interface
 if __name__ == "__main__":
-    gr.TabbedInterface(
-        [iface, gr.Interface(fn=lambda: "", inputs=None, outputs=merge_and_push_output)],
-        ["Generate Text", "Merge and Push"]
-    ).launch()
+    demo.launch()