liujch1998 commited on
Commit
b8a7cf9
Β·
1 Parent(s): 11a61b4

Update model name

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -17,7 +17,7 @@ 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/cd-pi'
21
  DATASET_REPO_URL = "https://huggingface.co/datasets/liujch1998/cd-pi-dataset"
22
  DATA_DIR = 'data'
23
  DATA_PATH = os.path.join(DATA_DIR, 'data.jsonl')
@@ -159,12 +159,13 @@ input_statement = gr.Dropdown(choices=examples, label='Statement:')
159
  input_model = gr.Textbox(label='Commonsense statement verification model:', value=MODEL_NAME, interactive=False)
160
  output = gr.outputs.Label(num_top_classes=2)
161
 
162
- description = '''This is a demo for a commonsense statement verification model. Under development.'''
 
163
 
164
  gr.Interface(
165
  fn=predict,
166
  inputs=[input_statement, input_model],
167
  outputs=output,
168
- title="cd-pi Demo",
169
  description=description,
170
  ).launch()
 
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"
22
  DATA_DIR = 'data'
23
  DATA_PATH = os.path.join(DATA_DIR, 'data.jsonl')
 
159
  input_model = gr.Textbox(label='Commonsense statement verification model:', value=MODEL_NAME, interactive=False)
160
  output = gr.outputs.Label(num_top_classes=2)
161
 
162
+ description = '''This is a demo for Vera, a commonsense statement verification model. Under development.
163
+ ⚠️ Data Collection: by default, we are collecting the inputs entered in this app to further improve and evaluate the model. Do not share any personal or sensitive information while using the app!'''
164
 
165
  gr.Interface(
166
  fn=predict,
167
  inputs=[input_statement, input_model],
168
  outputs=output,
169
+ title="Vera",
170
  description=description,
171
  ).launch()