moonahhyun commited on
Commit
383d625
1 Parent(s): 6812aa7

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +50 -0
app.py ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
3
+ from datasets import load_dataset
4
+
5
+ dataset = load_dataset('HUPD/hupd',
6
+ name='sample',
7
+ data_files="https://huggingface.co/datasets/HUPD/hupd/blob/main/hupd_metadata_2022-02-22.feather",
8
+ icpr_label=None,
9
+ train_filing_start_date='2016-01-01',
10
+ train_filing_end_date='2016-01-01',
11
+ val_filing_start_date='2016-01-30',
12
+ val_filing_end_date='2016-01-31',
13
+ )
14
+ p_number = dataset["validation"]["patent_number"][:10]
15
+ p_abstract = dataset["validation"]["abstract"][:10]
16
+ p_claims = dataset["validation"]["claims"][:10]
17
+ p_decision = dataset["validation"]["decision"][:10]
18
+
19
+ # Streamlit app
20
+ st.title("Patentability Score")
21
+ selected_model = st.selectbox("App. ID:", p_number)
22
+
23
+
24
+ # st.write("Select a patent application ID")
25
+ # text = st.text_input("Text:", "I love you!")
26
+ # # Prepare analysis model, tokenizer and pipeline
27
+ # def get_pipeline(selected_model):
28
+ # model = AutoModelForSequenceClassification.from_pretrained(selected_model)
29
+ # tokenizer = AutoTokenizer.from_pretrained(selected_model)
30
+ # pl = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
31
+ # return pl
32
+
33
+ # # Load the model and perform sentiment analysis
34
+ # if st.button("Submit"):
35
+ # with st.spinner("Analyzing sentiment..."):
36
+ # pl = get_pipeline(selected_model)
37
+ # result = pl(text)
38
+ # label = result[0]['label']
39
+ # if(selected_model == "cardiffnlp/twitter-roberta-base-sentiment"):
40
+ # if label == "LABEL_0": st.write("Sentiment: Negative")
41
+ # elif label == "LABEL_1": st.write("Sentiment: Neutral")
42
+ # elif label == "LABEL_2": st.write("Sentiment: Positive")
43
+ # elif(selected_model == "textattack/bert-base-uncased-SST-2"):
44
+ # if label == "LABEL_0": st.write("Sentiment: Negative")
45
+ # elif label == "LABEL_1": st.write("Sentiment: Positive")
46
+ # else:
47
+ # st.write(f"Sentiment: {label}")
48
+ # st.write(f"Confidence Score: {result[0]['score']:.2f}")
49
+ # else:
50
+ # st.write("Click 'Submit' for sentiment analysis.")