Databricks: No error but no output

[Meenakshi-10]

Hi, I tried running the model locally (using CPU only) but my memory ran out. So with the same source image I tried running it on databricks. I have stored the model file in DBFS. I updated the new model path in cog.yaml. There was no error, but it didn't perform the deblurring. After a bit of debugging, I noticed it stopped somewhere in the model.test() function. Any idea what's happening?