omer11a commited on
Commit
a55826f
·
1 Parent(s): 6c2e4ca

Added nltk resource download

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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",