Update app.py

app.py

@@ -462,6 +462,13 @@ scheduler = BackgroundScheduler()
 scheduler.add_job(func=background_next_image, trigger="interval", seconds=1)
 scheduler.start()
 
+@spaces.GPU()
+def encode_space(x):
+    im_emb, _ = pipe.encode_image(
+                image, DEVICE, 1, output_hidden_state
+            )
+    return im_emb
+
 # prep our calibration prompts
 for im in [
     './first.mp4',
@@ -476,9 +483,7 @@ for im in [
     tmp_df['paths'] = [im]
     image = list(imageio.imiter(im))
     image = image[len(image)//2]
-    im_emb, _ = pipe.encode_image(
-                image, DEVICE, 1, output_hidden_state
-            )
+    im_emb = encode_space(image)
 
     tmp_df['embeddings'] = [im_emb.detach().to('cpu')]
     tmp_df['user:rating'] = [{' ': ' '}]