Update app.py

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 import json
 from huggingface_hub import InferenceClient
 
-# Hugging Face model for maritime legal advice
-MODEL = "HuggingFaceH4/zephyr-7b-beta"
+# Use the fine-tuned maritime legal model
+MODEL = "nomiChroma3.1"
 
-# Initialize the client
+# Initialize the client with the correct model
 client = InferenceClient(MODEL)
 
 def respond(message, history: list[tuple[str, str]], system_message, max_tokens, temperature, top_p):
@@ -58,7 +58,7 @@ demo = gr.ChatInterface(
         gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
     ],
     title="Maritime Legal Compliance",
-    description="This chatbot uses a fine-tuned model for providing legal advice on Indian maritime law.",
+    description="This chatbot uses the fine-tuned model for providing legal advice on Indian maritime law.",
 )
 
 if __name__ == "__main__":