SyrWin commited on
Commit
2770f17
·
1 Parent(s): 7072c88

runable version

Browse files
Files changed (2) hide show
  1. README.md +0 -1
  2. app.py +2 -1
README.md CHANGED
@@ -5,7 +5,6 @@ colorFrom: yellow
5
  colorTo: green
6
  sdk: gradio
7
  sdk_version: 4.36.0
8
- python_version: 3.8
9
  app_file: app.py
10
  pinned: false
11
  ---
 
5
  colorTo: green
6
  sdk: gradio
7
  sdk_version: 4.36.0
 
8
  app_file: app.py
9
  pinned: false
10
  ---
app.py CHANGED
@@ -1,5 +1,7 @@
1
  import subprocess
2
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 
3
  import os
4
  import torch
5
  import argparse
@@ -15,7 +17,6 @@ from data_provider.data_utils import json_read, json_write
15
  from data_provider.data_utils import smiles2data, reformat_smiles
16
  import gradio as gr
17
  from datetime import datetime
18
- import spaces
19
 
20
  ## for pyg bug
21
  warnings.filterwarnings('ignore', category=UserWarning, message='TypedStorage is deprecated')
 
1
  import subprocess
2
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
3
+ subprocess.run('pip install -U timm', shell=True)
4
+ import spaces
5
  import os
6
  import torch
7
  import argparse
 
17
  from data_provider.data_utils import smiles2data, reformat_smiles
18
  import gradio as gr
19
  from datetime import datetime
 
20
 
21
  ## for pyg bug
22
  warnings.filterwarnings('ignore', category=UserWarning, message='TypedStorage is deprecated')