fffiloni commited on
Commit
0280a6c
·
1 Parent(s): d58b2b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -125,7 +125,7 @@ def infer():
125
  #frames, _, _ = read_video(str("./spacex.mp4"), output_format="TCHW")
126
  #print(f"FRAME BEFORE stack: {frames[100]}")
127
 
128
- prompt = "astronaut crew inside a spaceship, artwork by Claude Monet"
129
 
130
  pil2diff_img = Image.open("./frame1.jpg")
131
  canny_image = get_canny_filter(pil2diff_img)
@@ -147,8 +147,8 @@ def infer():
147
  #img1_batch = torch.stack([frames[0]])
148
  #img2_batch = torch.stack([frames[1]])
149
 
150
- img1_batch = torch.stack([input_diffused])
151
- img2_batch = torch.stack([input_frame_2])
152
 
153
  print(f"FRAME AFTER stack: {img1_batch}")
154
 
 
125
  #frames, _, _ = read_video(str("./spacex.mp4"), output_format="TCHW")
126
  #print(f"FRAME BEFORE stack: {frames[100]}")
127
 
128
+ prompt = "astronaut crew inside a spaceship, painting by Claude Monet"
129
 
130
  pil2diff_img = Image.open("./frame1.jpg")
131
  canny_image = get_canny_filter(pil2diff_img)
 
147
  #img1_batch = torch.stack([frames[0]])
148
  #img2_batch = torch.stack([frames[1]])
149
 
150
+ img1_batch = torch.stack([input_frame_2])
151
+ img2_batch = torch.stack([input_diffused])
152
 
153
  print(f"FRAME AFTER stack: {img1_batch}")
154