khanfou commited on
Commit
5d4ca87
1 Parent(s): 953ae4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -37,9 +37,9 @@ with st.form("patent-form"):
37
  tokenizer = AutoTokenizer.from_pretrained(model_name)
38
  classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
39
  abstract = df['abstract'].loc[df['patent_number'] == make_choice].astype("string")
40
- #X_train = df['id'].astype("string")
41
  #X_train = abstract.values.tolist()
42
- results = classifier(abstract, truncation=True)
43
  #result = hupd_model(make_choice)[0]
44
  score = result['score']
45
  st.write("The Patentability Score is:", score)
 
37
  tokenizer = AutoTokenizer.from_pretrained(model_name)
38
  classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
39
  abstract = df['abstract'].loc[df['patent_number'] == make_choice].astype("string")
40
+ X_train = abstract.astype("string")
41
  #X_train = abstract.values.tolist()
42
+ results = classifier(X_train, truncation=True)
43
  #result = hupd_model(make_choice)[0]
44
  score = result['score']
45
  st.write("The Patentability Score is:", score)