kitkeat commited on
Commit
ba3045f
1 Parent(s): b7e0b0f

Update app.py

Browse files

Added Dropdown to select discourse type

Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -9,10 +9,15 @@ tokenizer1 = AutoTokenizer.from_pretrained('kitkeat/distilbert-based-uncased-arg
9
  model1 = AutoModelForSequenceClassification.from_pretrained('kitkeat/distilbert-based-uncased-argumentativewriting',num_labels=3)
10
  #model2 = AutoModelForSequenceClassification.from_pretrained('kitkeat/bert-large-uncased-sparse-90-unstructured-pruneofa-argumentativewriting',num_labels=3)
11
  #model3 = AutoModelForSequenceClassification.from_pretrained('kitkeat/deberta-v3-base-argumentativewriting',num_labels=3)
 
 
 
12
  text = st.text_area('Input Here!')
13
 
14
  if text:
15
- inputs1 = tokenizer1(text, padding=True, truncation=True, return_tensors="pt")
 
 
16
  # inputs2 = tokenizer2(text, padding=True, truncation=True, return_tensors="pt")
17
  # inputs3 = tokenizer3(text, padding=True, truncation=True, return_tensors="pt")
18
 
9
  model1 = AutoModelForSequenceClassification.from_pretrained('kitkeat/distilbert-based-uncased-argumentativewriting',num_labels=3)
10
  #model2 = AutoModelForSequenceClassification.from_pretrained('kitkeat/bert-large-uncased-sparse-90-unstructured-pruneofa-argumentativewriting',num_labels=3)
11
  #model3 = AutoModelForSequenceClassification.from_pretrained('kitkeat/deberta-v3-base-argumentativewriting',num_labels=3)
12
+ option = st.selectbox(
13
+ 'Discourse Type',
14
+ ('Position', 'Concluding Statement', 'Claim', 'Counterclaim' , 'Evidence', 'Lead', 'Position', 'Rebuttal'))
15
  text = st.text_area('Input Here!')
16
 
17
  if text:
18
+ SEP = toknizer1.sep_token
19
+ text_edited = option + SEP + text
20
+ inputs1 = tokenizer1(text_edited , padding=True, truncation=True, return_tensors="pt")
21
  # inputs2 = tokenizer2(text, padding=True, truncation=True, return_tensors="pt")
22
  # inputs3 = tokenizer3(text, padding=True, truncation=True, return_tensors="pt")
23