Initial commit

README.md

@@ -1,8 +1,8 @@
 ---
 title: PaperClassifier
-emoji: 🐨
-colorFrom: green
-colorTo: pink
+emoji: 📝
+colorFrom: blue
+colorTo: gray
 sdk: streamlit
 sdk_version: 1.17.0
 app_file: app.py
@@ -10,4 +10,3 @@ pinned: false
 license: mit
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,26 @@
+from transformers import AutoTokenizer, AutoModelForMaskedLM, pipeline
+import streamlit as st
+
+
+@st.cache_data
+def prepare():
+    tokenizer = AutoTokenizer.from_pretrained("oracat/bert-paper-classifier")
+    model = AutoModelForSequenceClassification.from_pretrained("oracat/bert-paper-classifier")
+
+
+def process(text):
+    pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
+    result = pipe(text)[0]
+    return results['label']
+
+
+prepare()
+
+st.markdown("### Hello, paper classifier!")
+
+title = st.text_input("Enter the title...")
+abstract = st.text_area("... and maybe the abstract of the paper you want to classify")
+
+text = "\n".join([title, abstract])
+
+st.markdown(f"{process(text)}")

requirements.txt

@@ -0,0 +1,2 @@
+transformers
+torch