Update app.py

app.py

@@ -4,7 +4,7 @@ import spaces
 import torch
 import re
 
-model = PaliGemmaForConditionalGeneration.from_pretrained("gokaygokay/sd3-long-captioner").to("cuda").eval()
+model = PaliGemmaForConditionalGeneration.from_pretrained("gokaygokay/sd3-long-captioner").to("cpu").eval()
 processor = PaliGemmaProcessor.from_pretrained("gokaygokay/sd3-long-captioner")
 
 def modify_caption(caption: str) -> str:
@@ -28,13 +28,13 @@ def modify_caption(caption: str) -> str:
 
     return re.sub(pattern, replace_fn, caption, count=1, flags=re.IGNORECASE)
 
-@spaces.GPU
+
 def create_captions_rich(images):
     captions = []
     prompt = "caption en"
     
     for image in images:
-        model_inputs = processor(text=prompt, images=image, return_tensors="pt").to("cuda")
+        model_inputs = processor(text=prompt, images=image, return_tensors="pt").to("cpu")
         input_len = model_inputs["input_ids"].shape[-1]
         
         with torch.inference_mode():