Update app.py

app.py

@@ -574,18 +574,19 @@ scheduler.start()
 # TODO shouldn't call this before gradio launch, yeah?
 @spaces.GPU()
 def encode_space(x, im):
-    print('encode')
-    im_emb, _ = pipe.encode_image(
-                x, DEVICE, 1, output_hidden_state
-            )
-
-    print('encoded')
-    
-    print('pali_enc')
-    gemb = pali.vision_tower(im.to(dtype).to('cuda')).last_hidden_state
+    with torch.no_grad():
+        print('encode')
+        im_emb, _ = pipe.encode_image(
+                    x, DEVICE, 1, output_hidden_state
+                )
     
-    print('pali_enced')
-    return im_emb.to(torch.float32), gemb.to(torch.float32)
+        print('encoded')
+        
+        print('pali_enc')
+        gemb = pali.vision_tower(im.to(dtype).to('cuda')).last_hidden_state
+        
+        print('pali_enced')
+        return im_emb.to(torch.float32), gemb.to(torch.float32)
 
 demo.launch(share=True,)