fffiloni commited on
Commit
8337469
1 Parent(s): 25b5212

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -25,9 +25,9 @@ if torch.cuda.is_available():
25
 
26
  def pix2pix(
27
  prompt,
 
28
  text_guidance_scale,
29
  image_guidance_scale,
30
- image,
31
  steps,
32
  neg_prompt="",
33
  width=512,
@@ -125,9 +125,9 @@ def infer(prompt,video_in, seed_in, trim_value):
125
  print("set stop frames to: " + str(n_frame))
126
 
127
  for i in frames_list[0:int(n_frame)]:
128
- pix2pix_img = pix2pix(prompt,5.5,1.5,i,15,"",512,512,seed_in)
129
  print(pix2pix_img)
130
- image = Image.open(pix2pix_img)
131
  rgb_im = image.convert("RGB")
132
 
133
  # exporting the image
 
25
 
26
  def pix2pix(
27
  prompt,
28
+ image,
29
  text_guidance_scale,
30
  image_guidance_scale,
 
31
  steps,
32
  neg_prompt="",
33
  width=512,
 
125
  print("set stop frames to: " + str(n_frame))
126
 
127
  for i in frames_list[0:int(n_frame)]:
128
+ pix2pix_img = pix2pix(prompt,i,5.5,1.5,15,"",512,512,seed_in)
129
  print(pix2pix_img)
130
+ image = pix2pix_img
131
  rgb_im = image.convert("RGB")
132
 
133
  # exporting the image