khanfou commited on
Commit
ac50fe3
1 Parent(s): 60777d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -36,8 +36,12 @@ 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]
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]
 
36
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
37
  tokenizer = AutoTokenizer.from_pretrained(model_name)
38
  classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
39
+
40
+ #abstract = df['abstract'].loc[df['patent_number'] == make_choice]
41
+
42
+ decision = df['decision'].loc[df['patent_number'] == make_choice]
43
+ #X_train = abstract.to_string()
44
+ X_train = decision.to_string()
45
  #X_train = abstract.values.tolist()
46
  results = classifier(X_train, truncation=True)
47
  #result = hupd_model(make_choice)[0]