osanseviero commited on
Commit
08de1fc
1 Parent(s): 8444104

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +0 -2
pipeline.py CHANGED
@@ -15,8 +15,6 @@ class PreTrainedPipeline():
15
  try:
16
  self.model = BigGAN.from_pretrained(path)
17
  except (IOError, OSError):
18
- directory = "/data/corpora"
19
- shutil.rmtree(directory)
20
  nltk.download('wordnet')
21
  self.model = BigGAN.from_pretrained(path)
22
  self.truncation = 0.1
 
15
  try:
16
  self.model = BigGAN.from_pretrained(path)
17
  except (IOError, OSError):
 
 
18
  nltk.download('wordnet')
19
  self.model = BigGAN.from_pretrained(path)
20
  self.truncation = 0.1