Update app.py

app.py

@@ -3,7 +3,7 @@ from janus.janusflow.models import MultiModalityCausalLM, VLChatProcessor
 from PIL import Image
 from diffusers.models import AutoencoderKL
 import numpy as np
-import gradio as gr  # Import gradio for UI
+import gradio as gr
 
 # CUDA availability check
 cuda_device = 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -176,6 +176,7 @@ def generate_image(prompt,
         return [Image.fromarray(images[i]).resize((1024, 1024), Image.LANCZOS) for i in range(images.shape[0])]
 
 
+
 # Gradio interface
 with gr.Blocks(title="JanusFlow Medical Image Assistant") as demo:
     gr.Markdown(value="# Medical Image Understanding and Generation")
@@ -197,16 +198,20 @@ with gr.Blocks(title="JanusFlow Medical Image Assistant") as demo:
             examples=[
                  [
                   "What are the visible structures in this ultrasound?",
-                 "./ultrasound.jpeg"
+                  Image.open("ultrasound.jpeg"), # Load Directly
                 ],
                   [
                   "Identify abnormalities in the image.",
-                 "./cardiac_ultrasound.jpeg"
+                   Image.open("cardiac_ultrasound.jpeg"), # Load Directly
                   ],
                   [
                   "Describe the features and histological analysis in this image.",
-                 "./histology.jpeg"
+                  Image.open("histology.jpeg"), # Load Directly
                   ],
+                   [
+                   "What are the characteristics and analysis of this image?",
+                    Image.open("histology2.jpeg")
+                   ]
             ],
             inputs=[question_input, image_input],
         )
@@ -244,4 +249,4 @@ with gr.Blocks(title="JanusFlow Medical Image Assistant") as demo:
         outputs=image_output
     )
     
-demo.launch(share=True)
+demo.launch(share=False) # disabled share for HF Spaces