Update app.py

app.py

@@ -29,12 +29,13 @@ def split_image(input_image, num_splits=4):
 
     return output_images
 
-pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch_dtype)
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch_dtype).to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
 
 @spaces.GPU 
 def infer(prompt, seed=1, randomize_seed=False, num_inference_steps=28):
+    print('entered the function')
     prompt_template = f"A side by side 4 frame image showing high quality consecutive stills from a looped gif animation moving from left to right. The scene has motion. The stills are of {prompt}"
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)