Update app.py
Browse files
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,
|
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([
|
151 |
-
img2_batch = torch.stack([
|
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 |
|