fffiloni commited on
Commit
d489594
1 Parent(s): 4382d6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -26,16 +26,11 @@ def pix2pix(
26
  input_image: Image.Image,
27
  instruction: str,
28
  steps: int,
29
- randomize_seed: bool,
30
  seed: int,
31
- randomize_cfg: bool,
32
  text_cfg_scale: float,
33
  image_cfg_scale: float,
34
  ):
35
- seed = random.randint(0, 100000) if randomize_seed else seed
36
- text_cfg_scale = round(random.uniform(6.0, 9.0), ndigits=2) if randomize_cfg else text_cfg_scale
37
- image_cfg_scale = round(random.uniform(1.2, 1.8), ndigits=2) if randomize_cfg else image_cfg_scale
38
-
39
  width, height = input_image.size
40
  factor = 512 / max(width, height)
41
  factor = math.ceil(min(width, height) * factor / 64) * 64 / min(width, height)
@@ -119,7 +114,8 @@ def infer(prompt,video_in, seed_in, trim_value):
119
 
120
  for i in frames_list[0:int(n_frame)]:
121
  pil_i = Image.open(i)
122
- pix2pix_img = pix2pix(pil_i, prompt, 50, False, seed_in, False, 7.5, 1.5)
 
123
  print(pix2pix_img)
124
  image = Image.open(pix2pix_img)
125
  rgb_im = image.convert("RGB")
 
26
  input_image: Image.Image,
27
  instruction: str,
28
  steps: int,
 
29
  seed: int,
 
30
  text_cfg_scale: float,
31
  image_cfg_scale: float,
32
  ):
33
+
 
 
 
34
  width, height = input_image.size
35
  factor = 512 / max(width, height)
36
  factor = math.ceil(min(width, height) * factor / 64) * 64 / min(width, height)
 
114
 
115
  for i in frames_list[0:int(n_frame)]:
116
  pil_i = Image.open(i)
117
+ print(pil_i.Image)
118
+ pix2pix_img = pix2pix(pil_i.Image, prompt, 50, seed_in, 7.5, 1.5)
119
  print(pix2pix_img)
120
  image = Image.open(pix2pix_img)
121
  rgb_im = image.convert("RGB")