changed spaces.GPU position.

app.py

@@ -1,6 +1,6 @@
-import subprocess
-subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-subprocess.run('pip install -U timm', shell=True)
+# import subprocess
+# subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+# subprocess.run('pip install -U timm', shell=True)
 import spaces
 import os
 import torch
@@ -197,6 +197,7 @@ class InferenceRunner:
             graph_list.append(graph_item)
         return graph_list, input_text
 
+    @spaces.GPU
     @torch.no_grad()
     def predict(self, rxn_dict, temperature=1):
         graphs, prompt_tokens = self.tokenize(rxn_dict)
@@ -236,7 +237,6 @@ class InferenceRunner:
         input_prompt_tokens['is_mol_token'] = is_mol_token
         return graphs, input_prompt_tokens
 
-@spaces.GPU
 def main(args):
     device = torch.device('cuda')
     # model