From 39698bc6b3f426d39a0145d66401043dab01779b Mon Sep 17 00:00:00 2001
From: guojianzhu <guojianzhu@kuaishou.com>
Date: Fri, 5 Jul 2024 14:16:03 +0800
Subject: [PATCH] chore: update readme and remove timm

---
 readme.md        |  2 +-
 requirements.txt | 11 +++++------
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/readme.md b/readme.md
index b0bd120..f394e2c 100644
--- a/readme.md
+++ b/readme.md
@@ -102,7 +102,7 @@ python inference.py -h
 
 ### 4. Gradio interface
 
-We also provide a Gradio interface for a better experience. Please install `gradio` and then run `app.py`:
+We also provide a Gradio interface for a better experience, just run by:
 
 ```bash
 python app.py
diff --git a/requirements.txt b/requirements.txt
index 1d05b23..73dbda9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
---extra-index-url https://download.pytorch.org/whl/cu118 
-torch==2.3.0 
-torchvision==0.18.0 
-torchaudio==2.3.0 
+--extra-index-url https://download.pytorch.org/whl/cu118
+torch==2.3.0
+torchvision==0.18.0
+torchaudio==2.3.0
 
 numpy==1.26.4
 pyyaml==6.0.1
@@ -10,7 +10,6 @@ scipy==1.13.1
 imageio==2.34.2
 lmdb==1.4.1
 tqdm==4.66.4
-timm==1.0.7
 rich==13.7.1
 ffmpeg==1.4
 onnxruntime-gpu==1.18.0
@@ -20,4 +19,4 @@ albumentations==1.4.10
 matplotlib==3.9.0
 imageio-ffmpeg==0.5.1
 tyro==0.8.5
-gradio==4.37.1
\ No newline at end of file
+gradio==4.37.1