fixes

app.py

@@ -144,7 +144,7 @@ def generate_gpu(in_im_embs, prompt='the scene'):
 
 def generate(in_im_embs, prompt='the scene'):
     output, im_emb = generate_gpu(in_im_embs, prompt)
-    nsfw = maybe_nsfw(output.images[0])
+    nsfw = False#maybe_nsfw(output.images[0])
     
     name = str(uuid.uuid4()).replace("-", "")
     path = f"/tmp/{name}.png"
@@ -552,8 +552,7 @@ for im, txt in [ # TODO more movement
     ]:
     tmp_df = pd.DataFrame(columns=['paths', 'embeddings', 'ips', 'user:rating', 'text', 'gemb'])
     tmp_df['paths'] = [im]
-    image = list(imageio.imiter(im))
-    image = image[len(image)//2]
+    image = Image.open(im).convert('RGB')
     im_emb = encode_space(image)
     
     tmp_df['embeddings'] = [im_emb.detach().to('cpu')]