fix display bug

app.py

@@ -5,7 +5,7 @@ import os
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import spaces
 
-title = """        # Welcome to Tonic's 🌊 Osmosis Structure - Text to JSON Converter
+title = """        # 🙋🏻‍♂️Welcome to 🌟Tonic's 🌊 Osmosis Structure - Text to JSON Converter
 """
 description = """       
         Convert unstructured text into well-formatted JSON using the Osmosis Structure 0.6B model.
@@ -64,8 +64,9 @@ def load_model():
             torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
             device_map="auto" if torch.cuda.is_available() else None,
             trust_remote_code=True,
-            token=hf_token
-            )
+            token=hf_token,
+            use_auth_token=hf_token  # Backward compatibility
+        )
         
         print("✅ Osmosis Structure model loaded successfully!")
         return True
@@ -186,22 +187,26 @@ def text_to_json(input_text, max_tokens=512, temperature=0.6, top_p=0.95, top_k=
     except Exception as e:
         return f"❌ Error generating JSON: {str(e)}"
 
-# Create Gradio interface
 def create_demo():
+    # Fixed: Remove duplicate with gr.Blocks declaration
     with gr.Blocks(
-        title="Osmosis Structure - Text to JSON Converter",
-        theme=gr.themes.Monochrome()
+        title=title,
+        theme=gr.themes.Monochrome(),
+        css="""
+        .gradio-container {
+            max-width: 1200px !important;
+        }
+        """
     ) as demo:
-        with gr.Blocks(theme=gr.themes.Base()) as demo:
-            with gr.Row():
-                gr.Markdown(title)
-            with gr.Row():
-                with gr.Column(scale=1):
-                    with gr.Group():                    
-                        gr.Markdown(description)
-                with gr.Column(scale=1):
-                    with gr.Group():
-                        gr.Markdown(joinus)
+        # Header section
+        # gr.Markdown(title)
+        
+        # Info section
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.Markdown(description)
+            with gr.Column(scale=1):
+                gr.Markdown(joinus)
         
         with gr.Row():
             with gr.Column(scale=1):