space update

app.py

@@ -27,9 +27,7 @@ def load_model():
             num_heads=num_heads, 
             context_length=context_length, 
             num_layers=num_layers
-        )
-        print("✅ Model loaded successfully! vocab size:", len(u_model.vocab))
-        
+        )        
         # Load the trained weights if available
         model_path = "v1/u_model.pth"
 
@@ -155,7 +153,14 @@ demo = gr.ChatInterface(
         ),
     ],
     title="🤖 Usta Model Chat",
-    description="Chat with a custom transformer language model built from scratch! This model specializes in geographical knowledge including countries, capitals, and cities."
+    description="Chat with a custom transformer language model built from scratch! This model specializes in geographical knowledge including countries, capitals, and cities.",
+    examples=[
+        "the capital of france",
+        "tell me about spain", 
+        "what is the capital of united states",
+        "paris is in",
+        "germany and its capital"
+    ]
 )
 
 if __name__ == "__main__":