Update app.py

app.py

@@ -37,18 +37,7 @@ from urllib.request import urlretrieve
 
 from scipy.interpolate import LinearNDInterpolator
 from imageio import imread, imwrite
-from diffusers import StableDiffusionImg2ImgPipeline
 
-model_id = "runwayml/stable-diffusion-v1-5"
-pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
-
-def diffuse(source_img):  
-    source_img = Image.open(source_img)
-    prompt = "space crew in a spaceship by Claude Monet"
-    image = pipe(prompt, source_img).images[0]  
-    print(f"DIFFUSED IMG: {image}")
-    image.save("astronaut_rides_horse.png")
 
 def write_flo(flow, filename):
     """
@@ -85,8 +74,6 @@ def infer():
     #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}")