Update app.py

app.py

@@ -195,7 +195,8 @@ def compose_language_descriptions(prompt, guidance_scale):
 
   out_img = upsamples[0].permute(1,2,0)
   out_img = (out_img+1)/2
-  out_img = np.array(out_img.data.to('cpu'))
+  out_img = (out_img.detach().cpu() * 255.).to(torch.uint8)
+  out_img = out_img.numpy()
   return out_img
 
 # create model for CLEVR Objects
@@ -269,7 +270,8 @@ def compose_clevr_objects(prompt, guidance_scale):
     samples = sample(coordinates)
     out_img = samples[0].permute(1,2,0)
     out_img = (out_img+1)/2
-    out_img = np.array(out_img.data.to('cpu'))
+    out_img = (out_img.detach().cpu() * 255.).to(torch.uint8)
+    out_img = out_img.numpy()
     return out_img