DongfuJiang commited on
Commit
cdf945c
·
1 Parent(s): 2af0ed4
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import gradio as gr
2
  import sys
3
  import os
4
- os.environ["CUDA_VISIBLE_DEVICES"] = "7"
5
  from datasets import load_dataset
6
  from typing import List
7
 
@@ -22,11 +21,13 @@ for example in SHUFFLED_EXAMPLES_DATASET.take(100):
22
  CANDIDATE_EXAMPLES[example['instruction']+example['input']] = example['candidates']
23
 
24
  # Download ranker checkpoint
25
- os.system("ls -l /home/user/.local/lib/python3.10/site-packages/llm_blender")
26
- if not os.path.exists("pairranker-deberta-v3-large.zip"):
27
- os.system("gdown https://drive.google.com/uc?id=1EpvFu_qYY0MaIu0BAAhK-sYKHVWtccWg")
28
- if not os.path.exists("pairranker-deberta-v3-large"):
29
- os.system("unzip pairranker-deberta-v3-large.zip")
 
 
30
 
31
  # Load Blender
32
  import llm_blender
 
1
  import gradio as gr
2
  import sys
3
  import os
 
4
  from datasets import load_dataset
5
  from typing import List
6
 
 
21
  CANDIDATE_EXAMPLES[example['instruction']+example['input']] = example['candidates']
22
 
23
  # Download ranker checkpoint
24
+ print("Downloading pairranker-deberta-v3-large.zip")
25
+ os.system("gdown https://drive.google.com/uc?id=1EpvFu_qYY0MaIu0BAAhK-sYKHVWtccWg")
26
+ print("Downloaded pairranker-deberta-v3-large.zip")
27
+ print("Unzipping pairranker-deberta-v3-large.zip")
28
+ os.system("unzip pairranker-deberta-v3-large.zip")
29
+ print("Unzipped pairranker-deberta-v3-large.zip")
30
+ os.system("ls -l")
31
 
32
  # Load Blender
33
  import llm_blender