Warlord-K commited on
Commit
cdf9209
1 Parent(s): c5c9eb9
Files changed (1) hide show
  1. Videobook/Videobook.py +2 -2
Videobook/Videobook.py CHANGED
@@ -17,8 +17,8 @@ class Videobook:
17
  def generate_imgs(self, sentences, steps):
18
  imgs = []
19
  for sentence in sentences:
20
- sentence['pos'] = self.style + ' of ' + sentence['pos'] + ', ' + self.tags
21
- imgs.append(self.pipe.generate(prompt = sentence['pos'], negative_prompt = sentence['neg'] + " ,nude, disfigured, blurry", num_inference_steps = steps))
22
  return imgs
23
 
24
  def generate(self, story, api_key, style, tags, model, steps):
 
17
  def generate_imgs(self, sentences, steps):
18
  imgs = []
19
  for sentence in sentences:
20
+ sentence['pos'] = self.style + ' of ' + str(sentence['pos']) + ', ' + self.tags
21
+ imgs.append(self.pipe.generate(prompt = sentence['pos'], negative_prompt = str(sentence['neg']) + " ,nude, disfigured, blurry", num_inference_steps = steps))
22
  return imgs
23
 
24
  def generate(self, story, api_key, style, tags, model, steps):