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",
|