Add GPU to image caption generation

app.py

@@ -54,6 +54,8 @@ def base64_to_pil(img_base64):
     pil_image = Image.open(byte_stream)
     return pil_image
 
+# Caption generate function
+@spaces.GPU # Designed to be effect-free in non-ZeroGPU environments, ensuring compatibility across different setups.
 def captioner(image):
     #base64_image = image_to_base64_str(image)
     # The BLIP model expects a PIL image directly
@@ -62,7 +64,7 @@ def captioner(image):
     
     return result[0]['generated_text']
 
-# Generate function
+# Image generate function
 @spaces.GPU # Designed to be effect-free in non-ZeroGPU environments, ensuring compatibility across different setups.
 def generate(prompt, steps):  
     # Generate image with Stable Diffusion