Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import os
 from PIL import Image
 import torch
+import spaces
 
 from kimi_vl.serve.frontend import reload_javascript
 from kimi_vl.serve.utils import (
@@ -90,6 +91,7 @@ def get_prompt(conversation) -> str:
 
 
 @wrap_gen_fn
+@spaces.GPU()
 def predict(
     text,
     images,