teowu commited on
Commit
26b237e
·
verified ·
1 Parent(s): 7807e08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,6 +3,7 @@ import gradio as gr
3
  import os
4
  from PIL import Image
5
  import torch
 
6
 
7
  from kimi_vl.serve.frontend import reload_javascript
8
  from kimi_vl.serve.utils import (
@@ -90,6 +91,7 @@ def get_prompt(conversation) -> str:
90
 
91
 
92
  @wrap_gen_fn
 
93
  def predict(
94
  text,
95
  images,
 
3
  import os
4
  from PIL import Image
5
  import torch
6
+ import spaces
7
 
8
  from kimi_vl.serve.frontend import reload_javascript
9
  from kimi_vl.serve.utils import (
 
91
 
92
 
93
  @wrap_gen_fn
94
+ @spaces.GPU()
95
  def predict(
96
  text,
97
  images,