Spaces:

rynmurdock
/

generative_recsys

Sleeping

rynmurdock commited on Jun 2, 2024

Commit

18ec492

1 Parent(s): 340a151

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -566,7 +566,7 @@ scheduler.start()
 #thread.start()
 # TODO shouldn't call this before gradio launch, yeah?
-@spaces.GPU(duration=20)
 def encode_space(x):
     im_emb, _ = pipe.encode_image(
                 x, DEVICE, 1, output_hidden_state

 #thread.start()
 # TODO shouldn't call this before gradio launch, yeah?
+@spaces.GPU(duration=100)
 def encode_space(x):
     im_emb, _ = pipe.encode_image(
                 x, DEVICE, 1, output_hidden_state