Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
15710ee
1
Parent(s):
79cf446
fix: minor fix
Browse files
app.py
CHANGED
|
@@ -21,7 +21,7 @@
|
|
| 21 |
|
| 22 |
import base64
|
| 23 |
from io import BytesIO
|
| 24 |
-
|
| 25 |
import gradio as gr
|
| 26 |
import torch
|
| 27 |
from app_modules.gradio_utils import (
|
|
@@ -191,7 +191,7 @@ def get_prompt(conv) -> str:
|
|
| 191 |
else:
|
| 192 |
return conv.get_prompt
|
| 193 |
|
| 194 |
-
|
| 195 |
@wrap_gen_fn
|
| 196 |
def predict(
|
| 197 |
text,
|
|
|
|
| 21 |
|
| 22 |
import base64
|
| 23 |
from io import BytesIO
|
| 24 |
+
import spaces
|
| 25 |
import gradio as gr
|
| 26 |
import torch
|
| 27 |
from app_modules.gradio_utils import (
|
|
|
|
| 191 |
else:
|
| 192 |
return conv.get_prompt
|
| 193 |
|
| 194 |
+
@spaces.GPU
|
| 195 |
@wrap_gen_fn
|
| 196 |
def predict(
|
| 197 |
text,
|