feat:2.0

app.py

@@ -71,10 +71,6 @@ def load_model(hf_token):
             "google/gemma-2b",       # Try base 2B model next
             "google/gemma-7b-it",    # Try 7B instruction-tuned model
             "google/gemma-7b",       # Try base 7B model
-            # Fallback to completely different models if all Gemma models fail
-            "meta-llama/Llama-2-7b-chat-hf",
-            "facebook/opt-1.3b",
-            "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
         ]
         
         print(f"Attempting to load models with token starting with: {hf_token[:5]}...")
@@ -467,7 +463,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
                 """
             )
             
-            with gr.Box(elem_id="text_gen_box"):
+            with gr.Group(elem_id="text_gen_box"):
                 with gr.Row(equal_height=True):
                     with gr.Column(scale=1):
                         text_gen_type = gr.Radio(
@@ -597,7 +593,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
                 """
             )
             
-            with gr.Box(elem_id="brainstorm_box"):
+            with gr.Group(elem_id="brainstorm_box"):
                 with gr.Row(equal_height=True):
                     with gr.Column(scale=1):
                         brainstorm_category = gr.Dropdown(
@@ -660,7 +656,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
                 """
             )
             
-            with gr.Box(elem_id="content_box"):
+            with gr.Group(elem_id="content_box"):
                 with gr.Row(equal_height=True):
                     with gr.Column(scale=1):
                         content_type = gr.Dropdown(
@@ -730,7 +726,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
                 """
             )
             
-            with gr.Box(elem_id="email_box"):
+            with gr.Group(elem_id="email_box"):
                 with gr.Row(equal_height=True):
                     with gr.Column(scale=1):
                         email_type = gr.Dropdown(
@@ -794,7 +790,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
                 """
             )
             
-            with gr.Box(elem_id="edit_box"):
+            with gr.Group(elem_id="edit_box"):
                 with gr.Row(equal_height=True):
                     with gr.Column(scale=1):
                         edit_text = gr.Textbox(
@@ -845,7 +841,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
                 """
             )
             
-            with gr.Box(elem_id="explain_box"):
+            with gr.Group(elem_id="explain_box"):
                 with gr.Row(equal_height=True):
                     with gr.Column(scale=1):
                         explain_topic = gr.Textbox(
@@ -908,7 +904,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
                 """
             )
             
-            with gr.Box(elem_id="classify_box"):
+            with gr.Group(elem_id="classify_box"):
                 with gr.Row(equal_height=True):
                     with gr.Column(scale=1):
                         classify_text = gr.Textbox(
@@ -972,7 +968,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
                 """
             )
             
-            with gr.Box(elem_id="extract_box"):
+            with gr.Group(elem_id="extract_box"):
                 with gr.Row(equal_height=True):
                     with gr.Column(scale=1):
                         extract_text = gr.Textbox(
@@ -1038,7 +1034,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
             with gr.Tabs() as comprehension_tabs:
                 # Summarization
                 with gr.TabItem("Summarization", id="subtab_summarize"):
-                    with gr.Box(elem_id="summarize_box"):
+                    with gr.Group(elem_id="summarize_box"):
                         with gr.Row(equal_height=True):
                             with gr.Column(scale=1):
                                 summarize_text = gr.Textbox(
@@ -1062,7 +1058,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
                 
                 # Question Answering
                 with gr.TabItem("Question Answering", id="subtab_qa"):
-                    with gr.Box(elem_id="qa_box"):
+                    with gr.Group(elem_id="qa_box"):
                         with gr.Row(equal_height=True):
                             with gr.Column(scale=1):
                                 qa_text = gr.Textbox(
@@ -1091,7 +1087,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
                 
                 # Translation
                 with gr.TabItem("Translation", id="subtab_translate"):
-                    with gr.Box(elem_id="translate_box"):
+                    with gr.Group(elem_id="translate_box"):
                         with gr.Row(equal_height=True):
                             with gr.Column(scale=1):
                                 translate_text = gr.Textbox(
@@ -1132,7 +1128,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
             with gr.Tabs() as code_tabs:
                 # Code Generation
                 with gr.TabItem("Code Generation", id="subtab_code_gen"):
-                    with gr.Box(elem_id="code_gen_box"):
+                    with gr.Group(elem_id="code_gen_box"):
                         with gr.Row(equal_height=True):
                             with gr.Column(scale=1):
                                 code_language = gr.Dropdown(
@@ -1190,7 +1186,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
                 
                 # Code Explanation
                 with gr.TabItem("Code Explanation", id="subtab_code_explain"):
-                    with gr.Box(elem_id="code_explain_box"):
+                    with gr.Group(elem_id="code_explain_box"):
                         with gr.Row(equal_height=True):
                             with gr.Column(scale=1):
                                 code_to_explain = gr.Code(
@@ -1225,7 +1221,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
                 
                 # Code Debugging
                 with gr.TabItem("Code Debugging", id="subtab_code_debug"):
-                    with gr.Box(elem_id="code_debug_box"):
+                    with gr.Group(elem_id="code_debug_box"):
                         with gr.Row(equal_height=True):
                             with gr.Column(scale=1):
                                 code_to_debug = gr.Code(
@@ -1259,7 +1255,7 @@ with gr.Blocks(theme=GemmaLightTheme()) as demo:
                     )
     
     # Footer
-    with gr.Box(elem_id="footer"):
+    with gr.Group(elem_id="footer"):
         gr.Markdown(
             """
             ## About Gemma