ayethuzar commited on
Commit
00a1bcd
·
unverified ·
1 Parent(s): 9cbaf8e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -51,11 +51,11 @@ 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
- st.write(test)
56
 
57
  if submit:
58
- batch = tokenizer(test, padding = True, truncation = True, max_length = 512, return_tensors = "pt")
59
 
60
  with torch.no_grad():
61
  outputs = model(**batch)
 
51
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
52
  tokenizer = AutoTokenizer.from_pretrained(model_name)
53
 
54
+ text = [user_input_abstract[0] + user_input_claims[0]]
55
+ st.success(f'text: {text}')
56
 
57
  if submit:
58
+ batch = tokenizer(text, padding = True, truncation = True, max_length = 512, return_tensors = "pt")
59
 
60
  with torch.no_grad():
61
  outputs = model(**batch)