khanfou commited on
Commit
39ae039
1 Parent(s): 5d4ca87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,8 +36,8 @@ with st.form("patent-form"):
36
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
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]
 
36
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
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]
40
+ X_train = abstract.to_string()
41
  #X_train = abstract.values.tolist()
42
  results = classifier(X_train, truncation=True)
43
  #result = hupd_model(make_choice)[0]