liujch1998 commited on
Commit
2600399
β€’
1 Parent(s): 7128b84
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
15
 
16
  HF_TOKEN_DOWNLOAD = os.environ['HF_TOKEN_DOWNLOAD']
17
  HF_TOKEN_UPLOAD = os.environ['HF_TOKEN_UPLOAD']
18
- MODE = os.environ['MODE']
19
 
20
  MODEL_NAME = 'liujch1998/vera'
21
  DATASET_REPO_URL = "https://huggingface.co/datasets/liujch1998/cd-pi-dataset"
@@ -203,7 +203,7 @@ with gr.Blocks() as demo:
203
  submit = gr.Button(label='Submit', variant='primary')
204
  with gr.Column(scale=1):
205
  output = gr.outputs.Label(num_top_classes=2)
206
- output_raw = gr.outputs.JSON(visible=True)
207
  with gr.Row():
208
  feedback_agree = gr.Button(label='πŸ‘ Agree', variant='primary')
209
  feedback_disagree = gr.Button(label='πŸ‘Ž Disagree', variant='primary')
 
15
 
16
  HF_TOKEN_DOWNLOAD = os.environ['HF_TOKEN_DOWNLOAD']
17
  HF_TOKEN_UPLOAD = os.environ['HF_TOKEN_UPLOAD']
18
+ MODE = os.environ['MODE'] # 'debug' or 'prod'
19
 
20
  MODEL_NAME = 'liujch1998/vera'
21
  DATASET_REPO_URL = "https://huggingface.co/datasets/liujch1998/cd-pi-dataset"
 
203
  submit = gr.Button(label='Submit', variant='primary')
204
  with gr.Column(scale=1):
205
  output = gr.outputs.Label(num_top_classes=2)
206
+ output_raw = gr.outputs.JSON()
207
  with gr.Row():
208
  feedback_agree = gr.Button(label='πŸ‘ Agree', variant='primary')
209
  feedback_disagree = gr.Button(label='πŸ‘Ž Disagree', variant='primary')