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')]