ayethuzar commited on
Commit
e904e4b
·
unverified ·
1 Parent(s): 9b8e2c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -42,16 +42,16 @@ option = st.selectbox('Which other sections would you like to view?', tuple_of_c
42
 
43
  st.write('You selected:', option)
44
 
 
 
45
  form = st.form(key='abstract-claims-form')
46
- user_input_abstract = st.text_area(label = 'abstract', value = dataframe['abstract'][0][0])
47
- user_input_claims = st.text_area(label = 'claims', value = dataframe['claims'][0][0])
48
  submit = form.form_submit_button('Submit')
49
 
50
  model_name = "ayethuzar/tuned-for-patentability"
51
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
52
  tokenizer = AutoTokenizer.from_pretrained(model_name)
53
 
54
- test = [user_input_abstract + user_input_claims]
55
 
56
  if submit:
57
  batch = tokenizer(test, padding = True, truncation = True, max_length = 512, return_tensors = "pt")
 
42
 
43
  st.write('You selected:', option)
44
 
45
+ user_input_abstract = st.text_area(label = 'abstract', value = dataframe['abstract'][0])
46
+ user_input_claims = st.text_area(label = 'claims', value = dataframe['claims'][0])
47
  form = st.form(key='abstract-claims-form')
 
 
48
  submit = form.form_submit_button('Submit')
49
 
50
  model_name = "ayethuzar/tuned-for-patentability"
51
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
52
  tokenizer = AutoTokenizer.from_pretrained(model_name)
53
 
54
+ test = [user_input_abstract[0] + user_input_claims[0]
55
 
56
  if submit:
57
  batch = tokenizer(test, padding = True, truncation = True, max_length = 512, return_tensors = "pt")