publichealthsurveillance commited on
Commit
49c795b
1 Parent(s): 2b0cd0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -6,6 +6,8 @@ examples = [
6
  ["I only get my kids the ones I got....I've turned down many so called 'vaccines'"],
7
  ["In child protective services, further providing for definitions, for immunity from liability"],
8
  ["Lol what? Measles is a real thing. Get vaccinated"]]
 
 
9
 
10
  predictor_bert = ktrain.load_predictor('bert')
11
  predictor_mental = ktrain.load_predictor('mentalbert')
@@ -28,6 +30,7 @@ mental_io = gr.Interface(fn=mentalbert, inputs="text", outputs="text")
28
  phs_io = gr.Interface(fn=phsbert, inputs="text", outputs="text")
29
 
30
  Parallel(bert_io, mental_io, phs_io,
31
- examples=examples, title="Vaccine Sentiment Task - VS2",
32
- description="Enter vaccine-related tweets to generate sentiment from 3 models (BERT, MentalBERT, PHS-BERT). Label 0='vaccine critical', 1='neutral', 2='vaccine supportive'. The three provided examples are labelled 0,1,2 respectively. For details about VS2, refer to our paper (linked provided in https://huggingface.co/publichealthsurveillance/PHS-BERT).",
 
33
  theme="peach").launch()
 
6
  ["I only get my kids the ones I got....I've turned down many so called 'vaccines'"],
7
  ["In child protective services, further providing for definitions, for immunity from liability"],
8
  ["Lol what? Measles is a real thing. Get vaccinated"]]
9
+ title = "Vaccine Sentiment Task - VS2"
10
+ desc = "Enter vaccine-related tweets to generate sentiment from 3 models (BERT, MentalBERT, PHS-BERT). Label 0='vaccine critical', 1='neutral', 2='vaccine supportive'. The three provided examples are labelled 0,1,2 respectively. For details about VS2, refer to our paper (linked provided in https://huggingface.co/publichealthsurveillance/PHS-BERT)."
11
 
12
  predictor_bert = ktrain.load_predictor('bert')
13
  predictor_mental = ktrain.load_predictor('mentalbert')
 
30
  phs_io = gr.Interface(fn=phsbert, inputs="text", outputs="text")
31
 
32
  Parallel(bert_io, mental_io, phs_io,
33
+ examples=examples,
34
+ title=title,
35
+ description=desc,
36
  theme="peach").launch()