Update app.py

app.py

@@ -12,20 +12,30 @@ model_name_or_path = 'GoodBaiBai88/M3D-LaMed-Phi-3-4B'
 proj_out_num = 256
 
 # Load model and tokenizer
-model = AutoModelForCausalLM.from_pretrained(
-    model_name_or_path,
-    torch_dtype=dtype,
-    device_map=device,
-    trust_remote_code=True
-)
+try:
+    model = AutoModelForCausalLM.from_pretrained(
+        model_name_or_path,
+        torch_dtype=dtype,
+        device_map=device,
+        trust_remote_code=True
+    )
+    print("Model loaded successfully!")
+except Exception as e:
+    print(f"Error loading model: {e}")
+    raise
 
-tokenizer = AutoTokenizer.from_pretrained(
-    model_name_or_path,
-    model_max_length=512,
-    padding_side="right",
-    use_fast=False,
-    trust_remote_code=True
-)
+try:
+    tokenizer = AutoTokenizer.from_pretrained(
+        model_name_or_path,
+        model_max_length=512,
+        padding_side="right",
+        use_fast=False,
+        trust_remote_code=True
+    )
+    print("Tokenizer loaded successfully!")
+except Exception as e:
+    print(f"Error loading tokenizer: {e}")
+    raise
 
 # Chat history storage
 chat_history = []
@@ -117,10 +127,7 @@ def clear_chat():
 
 # Gradio UI
 with gr.Blocks(theme=gr.themes.Soft()) as chat_ui:
-    # Header Section
-    with gr.Row():
-        gr.Markdown("# 🏥 Medical Image Analysis Chatbot")
-        search_bar = gr.Textbox(label="Search", placeholder="Search...", interactive=True)
+    gr.Markdown("# 🏥 Medical Image Analysis Chatbot")
     
     # File upload section
     with gr.Row():