Spaces:
Runtime error
Runtime error
Update app.py
Browse filesAdded Dropdown to select discourse type
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 |
-
|
|
|
|
|
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 |
|