Hugging Face's logo

Spaces:
amsterdamNLP
/
CLIP-attention-rollout
Runtime error

App Files Community
paul hilders commited on
Commit
2a620ca
·
1 Parent(s): a676926

Import spacy module

Browse files
Files changed (2) hide show
  1. app.py +3 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import torch
8
  import CLIP.clip as clip
9
 
10
  import spacy
11
- from spacy import displacy
12
 
13
 
14
  from clip_grounding.utils.image import pad_to_square
@@ -25,7 +25,8 @@ clip.clip._MODELS = {
25
  device = "cuda" if torch.cuda.is_available() else "cpu"
26
  model, preprocess = clip.load("ViT-B/32", device=device, jit=False)
27
 
28
- nlp = spacy.load("en_core_web_sm")
 
29
 
30
  # Gradio Section:
31
  def run_demo(image, text):
 
8
  import CLIP.clip as clip
9
 
10
  import spacy
11
+ import en_core_web_sm
12
 
13
 
14
  from clip_grounding.utils.image import pad_to_square
 
25
  device = "cuda" if torch.cuda.is_available() else "cpu"
26
  model, preprocess = clip.load("ViT-B/32", device=device, jit=False)
27
 
28
+ # nlp = spacy.load("en_core_web_sm")
29
+ nlp = en_core_web_sm.load()
30
 
31
  # Gradio Section:
32
  def run_demo(image, text):
requirements.txt CHANGED
@@ -98,7 +98,7 @@ six==1.16.0
98
  sniffio==1.2.0
99
  soupsieve==2.3.2.post1
100
  spacy
101
- https://huggingface.co/spacy/en_core_web_sm/resolve/main/en_core_web_sm-any-py3-none-any.whl
102
  stack-data==0.3.0
103
  tangled-up-in-unicode==0.2.0
104
  termcolor==1.1.0
 
98
  sniffio==1.2.0
99
  soupsieve==2.3.2.post1
100
  spacy
101
+ en_core_web_sm
102
  stack-data==0.3.0
103
  tangled-up-in-unicode==0.2.0
104
  termcolor==1.1.0