Update app.py

app.py

@@ -195,7 +195,7 @@ def compose_language_descriptions(prompt, guidance_scale):
 
   out_img = upsamples[0].permute(1,2,0)
   out_img = (out_img+1)/2
-  out_img = (out_img.detach().cpu() * 255.).to(torch.uint8)
+  out_img = (out_img.detach().cpu() * 255.).to(th.uint8)
   out_img = out_img.numpy()
   return out_img
 
@@ -270,7 +270,7 @@ 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 = (out_img.detach().cpu() * 255.).to(torch.uint8)
+    out_img = (out_img.detach().cpu() * 255.).to(th.uint8)
     out_img = out_img.numpy()
     return out_img