Update app.py

app.py

@@ -3,12 +3,13 @@ from lmdeploy.vl import load_image
 import spaces
 import gradio as gr
 from PIL import Image
+import numpy as np
 
-
+pipe = pipeline('gokaygokay/llava-llama3-docci')
 
 @spaces.GPU
 def create_captions_llava_llama3_docci(image):
-    pipe = pipeline('gokaygokay/llava-llama3-docci')
+    
     gen_config = GenerationConfig(repetition_penalty=1.10)
     image = Image.fromarray(np.uint8(image)).convert('RGB')
     response = pipe(('describe this image in detail', image), gen_config=gen_config)
@@ -33,7 +34,7 @@ with gr.Blocks(css=css) as demo:
                 output = gr.Text(label="Caption")
             
         gr.Examples(
-        [["assets/image1.png"], ["assets/image2.PNG"], ["assets/image3.jpg"]],
+        [["image1.jpg"], ["image2.jpg"], ["image3.png"]],
         inputs = [input_img],
         outputs = [output],
         fn=create_captions_llava_llama3_docci,