Spaces:

rynmurdock
/

generative_recsys

Running on L40S

rynmurdock commited on Jun 2, 2024

Commit

1b78d70

1 Parent(s): d0f63f0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -568,7 +568,7 @@ scheduler.start()
 @spaces.GPU()
 def encode_space(x):
     im_emb, _ = pipe.encode_image(
-                image, DEVICE, 1, output_hidden_state
             )
     im = torchvision.transforms.ToTensor()(x).unsqueeze(0)

 @spaces.GPU()
 def encode_space(x):
     im_emb, _ = pipe.encode_image(
+                x, DEVICE, 1, output_hidden_state
             )
     im = torchvision.transforms.ToTensor()(x).unsqueeze(0)