Update app.py

app.py

@@ -185,7 +185,7 @@ def complete_address_interface(partial_address):
         return "❌ Model not loaded. Please check the logs."
     
     result = llama_system.complete_partial_address(partial_address)
-    return f"**Partial Address:** {partial_address}\n\n**Completed Address:**\n{result}"
+    return f"**Partial Address:** {partial_address}\n\n**Completed Address:**\n{result}\n\n*⚠️ Note: This feature has limited training data and results may vary in quality.*"
 
 def standardize_address_interface(address_text):
     """Interface function for address standardization"""
@@ -193,7 +193,7 @@ def standardize_address_interface(address_text):
         return "❌ Model not loaded. Please check the logs."
     
     result = llama_system.standardize_address(address_text)
-    return f"**Original:** {address_text}\n\n**Standardized:**\n{result}"
+    return f"**Original:** {address_text}\n\n**Standardized:**\n{result}\n\n*⚠️ Note: This feature has limited training data and results may vary in quality.*"
 
 # Sample data
 sample_addresses = [
@@ -224,13 +224,16 @@ with gr.Blocks(title="Llama Address Intelligence", theme=gr.themes.Soft()) as de
     gr.Markdown("""
     # 🦙 Llama 3.2-1B Address Intelligence
     
-    Powered by a fine-tuned Llama 3.2-1B model specialized for Indian address processing. This lightweight model can extract components, complete partial addresses, and standardize informal address formats.
+    Powered by a fine-tuned Llama 3.2-1B model specialized for Indian address processing. 
+    
+    **⭐ Best Performance**: Entity extraction from complete addresses  
+    **⚠️ Limited Performance**: Address completion and standardization (limited training data)
     
     **Model:** [shiprocket-ai/open-llama-1b-address-completion](https://huggingface.co/shiprocket-ai/open-llama-1b-address-completion)
     """)
     
     with gr.Tab("📋 Extract Components"):
-        gr.Markdown("Extract structured components from complete addresses")
+        gr.Markdown("⭐ **BEST PERFORMANCE** - Extract structured components from complete addresses")
         with gr.Row():
             with gr.Column(scale=1):
                 extract_input = gr.Textbox(
@@ -265,7 +268,7 @@ with gr.Blocks(title="Llama Address Intelligence", theme=gr.themes.Soft()) as de
         )
     
     with gr.Tab("✨ Complete Address"):
-        gr.Markdown("Complete partial or incomplete addresses using AI")
+        gr.Markdown("⚠️ **EXPERIMENTAL** - Complete partial addresses (limited training data - results may vary)")
         with gr.Row():
             with gr.Column(scale=1):
                 complete_input = gr.Textbox(
@@ -300,7 +303,7 @@ with gr.Blocks(title="Llama Address Intelligence", theme=gr.themes.Soft()) as de
         )
     
     with gr.Tab("📐 Standardize Format"):
-        gr.Markdown("Convert informal or messy addresses into proper standardized format")
+        gr.Markdown("⚠️ **EXPERIMENTAL** - Convert informal addresses to standardized format (limited training data - results may vary)")
         with gr.Row():
             with gr.Column(scale=1):
                 standardize_input = gr.Textbox(
@@ -363,6 +366,13 @@ with gr.Blocks(title="Llama Address Intelligence", theme=gr.themes.Soft()) as de
         - **Road Names**: Streets, lanes, main roads
         - **Landmarks**: Notable reference points
         
+        ### Performance Notes
+        - **⭐ Entity Extraction**: Excellent performance - primary use case
+        - **⚠️ Address Completion**: Limited training data - experimental feature
+        - **⚠️ Format Standardization**: Limited training data - experimental feature
+        
+        **Recommendation**: Use this model primarily for address component extraction.
+        
         ### Use Cases
         - **E-commerce**: Auto-complete checkout addresses
         - **Forms**: Intelligent address suggestions