runable version

README.md

@@ -5,7 +5,6 @@ colorFrom: yellow
 colorTo: green
 sdk: gradio
 sdk_version: 4.36.0
-python_version: 3.8
 app_file: app.py
 pinned: false
 ---

app.py

@@ -1,5 +1,7 @@
 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
 import argparse
@@ -15,7 +17,6 @@ from data_provider.data_utils import json_read, json_write
 from data_provider.data_utils import smiles2data, reformat_smiles
 import gradio as gr
 from datetime import datetime
-import spaces
 
 ## for pyg bug
 warnings.filterwarnings('ignore', category=UserWarning, message='TypedStorage is deprecated')