Update app.py

app.py

@@ -1,11 +1,15 @@
 import gradio as gr
 import torch
 import numpy as np
-from transformers import AutoModelForCausalLM
+from transformers import AutoModelForCausalLM, AutoTokenizer
 from janus.models import VLChatProcessor
 from PIL import Image
 import spaces
 
+# Suppress specific warnings
+import warnings
+warnings.filterwarnings("ignore", category=FutureWarning)
+
 # Medical Imaging Analysis Configuration
 MEDICAL_CONFIG = {
     "echo_guidelines": "ASE 2023 Standards",
@@ -38,6 +42,9 @@ if torch.cuda.is_available():
 
 vl_chat_processor = VLChatProcessor.from_pretrained(model_path)
 
+# **Fix: Set legacy=False in tokenizer to use the new behavior**
+vl_chat_processor.tokenizer = AutoTokenizer.from_pretrained(model_path, legacy=False)
+
 # Medical Image Processing Pipelines
 def preprocess_echo(image):
     """Process echocardiography images"""
@@ -147,11 +154,11 @@ with gr.Blocks(title="Cardiac & Histopathology AI", theme=gr.themes.Soft()) as d
         label="Example Medical Cases"
     )
 
-    @demo.func
-    def analyze_and_display(image, clinical_context, modality):
-        report = analyze_medical_case(image, clinical_context, modality)
-        return report
-    
-    analyze_btn.click(analyze_and_display, [image_input, clinical_input, modality_select], report_output)
+    # **Fixed: Removed @demo.func and used .click() correctly**
+    analyze_btn.click(
+        analyze_medical_case,
+        [image_input, clinical_input, modality_select],
+        report_output
+    )
 
 demo.launch(share=True)