Spaces:
Runtime error
Runtime error
Added nltk resource download
Browse files
app.py
CHANGED
@@ -7,6 +7,7 @@ from pytorch_lightning import seed_everything
|
|
7 |
import spaces
|
8 |
import gradio as gr
|
9 |
import torch
|
|
|
10 |
import numpy as np
|
11 |
from PIL import Image, ImageDraw
|
12 |
|
@@ -196,6 +197,8 @@ def main():
|
|
196 |
model.enable_xformers_memory_efficient_attention()
|
197 |
model.enable_sequential_cpu_offload()
|
198 |
|
|
|
|
|
199 |
with gr.Blocks(
|
200 |
css=css,
|
201 |
title="Bounded Attention demo",
|
|
|
7 |
import spaces
|
8 |
import gradio as gr
|
9 |
import torch
|
10 |
+
import nltk
|
11 |
import numpy as np
|
12 |
from PIL import Image, ImageDraw
|
13 |
|
|
|
197 |
model.enable_xformers_memory_efficient_attention()
|
198 |
model.enable_sequential_cpu_offload()
|
199 |
|
200 |
+
nltk.download('averaged_perceptron_tagger')
|
201 |
+
|
202 |
with gr.Blocks(
|
203 |
css=css,
|
204 |
title="Bounded Attention demo",
|