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'] = [{' ': ' '}]