Update app.py

app.py

@@ -196,7 +196,7 @@ def generate(in_im_embs, prompt='the scene'):
     if nsfw:
         gr.Warning("NSFW content detected.")
         # TODO could return an automatic dislike of auto dislike on the backend for neither as well; just would need refactoring.
-        return None, im_emb
+        return None, im_emb, gemb
     
     
     output.frames[0] = output.frames[0] + list(reversed(output.frames[0]))
@@ -587,7 +587,7 @@ def encode_space(x, im):
         gemb = pali.vision_tower(im.to(dtype).to('cuda')).last_hidden_state
         
         print('pali_enced')
-        return im_emb.to(torch.float32).to('cpu'), gemb.to(torch.float32).to('cpu')
+        return im_emb, gemb
 
 demo.launch(share=True,)