Update app.py

app.py

@@ -39,7 +39,7 @@ from scipy.interpolate import LinearNDInterpolator
 from imageio import imread, imwrite
 from diffusers import StableDiffusionImg2ImgPipeline
 
-def diffuse(source_img, prompt):
+def diffuse(source_img):
     model_id = "runwayml/stable-diffusion-v1-5"
     pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
     pipe = pipe.to("cuda")
@@ -81,8 +81,11 @@ def infer():
     #video_path = Path(tempfile.mkdtemp()) / "basketball.mp4"
     #_ = urlretrieve(video_url, video_path)
    
-    frames, _, _ = read_video(str("./spacex.mp4"), output_format="TCHW")
-    print(f"FRAME BEFORE stack: {frames[100]}")
+    #frames, _, _ = read_video(str("./spacex.mp4"), output_format="TCHW")
+    #print(f"FRAME BEFORE stack: {frames[100]}")
+
+    #testing difusion working properly
+    diffuse("./frame1.jpg")
     
     input_frame_1 = read_image(str("./frame1.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_1}")