DongfuJiang commited on
Commit
19d0707
·
1 Parent(s): ca4f279
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  import sys
3
  import os
 
4
  from datasets import load_dataset
5
  from typing import List
6
 
@@ -29,9 +30,10 @@ for example in SHUFFLED_EXAMPLES_DATASET.take(100):
29
 
30
  # Download ranker checkpoint
31
  if not os.path.exists("pairranker-deberta-v3-large.zip"):
32
- os.system("gdown https://drive.google.com/uc?id=1EpvFu_qYY0MaIu0BAAhK-sYKHVWtccWg")
33
  if not os.path.exists("pairranker-deberta-v3-large"):
34
- os.system("unzip pairranker-deberta-v3-large.zip")
 
35
 
36
  # Load Blender
37
  import llm_blender
 
1
  import gradio as gr
2
  import sys
3
  import os
4
+ import zipfile
5
  from datasets import load_dataset
6
  from typing import List
7
 
 
30
 
31
  # Download ranker checkpoint
32
  if not os.path.exists("pairranker-deberta-v3-large.zip"):
33
+ os.system("gdown https://drive.google.com/uc?id=1EpvFu_qYY0MaIu0BAAhK-sYKHVWtccWg")
34
  if not os.path.exists("pairranker-deberta-v3-large"):
35
+ with zipfile.ZipFile("pairranker-deberta-v3-large.zip", 'r') as zip_ref:
36
+ zip_ref.extractall(".")
37
 
38
  # Load Blender
39
  import llm_blender