Update app.py
Browse files
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,
|