bug fix

inference.py

@@ -43,9 +43,10 @@ def inference():
   imageAPI = DiffusionImageAPI(diffusion)
 
   images, versions = diffusion.sample(1)
-  images = []
-  for image in versions:
-    images.append(imageAPI.tensor_to_image(image.squeeze(0)))
+  #images = []
+  #for image in versions:
+  #  images.append(imageAPI.tensor_to_image(image.squeeze(0)))
+
   
   #print(len(images))
   #print(images[0])
@@ -55,7 +56,7 @@ def inference():
   #               append_images=images[1:],
   #               duration=100,
   #               loop=0)
-  return images[-1]
+  return imageAPI.tensor_to_image(images.squeeze(0))
 
 if __name__ == "__main__":
   inference().show()