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,