Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -47,6 +47,7 @@ tokenizer = transformers.AutoTokenizer.from_pretrained(
|
|
| 47 |
model.eval()
|
| 48 |
conv_mode = "v1"
|
| 49 |
|
|
|
|
| 50 |
@torch.inference_mode()
|
| 51 |
def inference_fn(
|
| 52 |
image,
|
|
@@ -130,7 +131,6 @@ def inference_fn(
|
|
| 130 |
|
| 131 |
return output
|
| 132 |
|
| 133 |
-
@spaces.GPU
|
| 134 |
with gr.Blocks() as demo:
|
| 135 |
gr.Markdown("# LLaVA-JP Demo")
|
| 136 |
|
|
|
|
| 47 |
model.eval()
|
| 48 |
conv_mode = "v1"
|
| 49 |
|
| 50 |
+
@spaces.GPU
|
| 51 |
@torch.inference_mode()
|
| 52 |
def inference_fn(
|
| 53 |
image,
|
|
|
|
| 131 |
|
| 132 |
return output
|
| 133 |
|
|
|
|
| 134 |
with gr.Blocks() as demo:
|
| 135 |
gr.Markdown("# LLaVA-JP Demo")
|
| 136 |
|