Merge remote-tracking branch 'origin/main'

app.py

@@ -1,5 +1,4 @@
 import os
-
 import pandas as pd
 
 os.system('cd fairseq;'
@@ -26,8 +25,9 @@ import gradio as gr
 # Register refcoco task
 tasks.register_task('ocr', OcrTask)
 
-os.system('wget https://shuangqing-multimodal.oss-cn-zhangjiakou.aliyuncs.com/ocr_general_clean.pt; '
-          'mkdir -p checkpoints; mv ocr_general_clean.pt checkpoints/ocr_general_clean.pt')
+if not os.path.exists("checkpoints/ocr_general_clean.pt"):
+    os.system('wget https://shuangqing-multimodal.oss-cn-zhangjiakou.aliyuncs.com/ocr_general_clean.pt; '
+              'mkdir -p checkpoints; mv ocr_general_clean.pt checkpoints/ocr_general_clean.pt')
 
 # turn on cuda if GPU is available
 use_cuda = torch.cuda.is_available()