Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| st.markdown("## Paper classification (arxiv.org taxonomy)") | |
| title = st.text_area("TITLE <required>") | |
| summary = st.text_area("SUMMARY <optional>") | |
| if (title == ''): | |
| raise AttributeError('The TITLE field is required! Please, fill it to see predictions.') | |
| text = title | |
| model_path = './model1_state_dict' | |
| if (summary != ''): | |
| text += '\n\n' + summary | |
| model_path = './model2_state_dict' | |
| terms_list = ['Artificial Intelligence (cs.AI)', 'Hardware Architecture (cs.AR)', 'Computational Complexity (cs.CC)', 'Computational Engineering, Finance, and Science (cs.CE)', 'Computational Geometry (cs.CG)', 'Computation and Language (cs.CL)', 'Cryptography and Security (cs.CR)', 'Computer Vision and Pattern Recognition (cs.CV)', 'Computers and Society (cs.CY)', 'Databases (cs.DB)', 'Distributed, Parallel, and Cluster Computing (cs.DC)', 'Digital Libraries (cs.DL)', 'Discrete Mathematics (cs.DM)', 'Data Structures and Algorithms (cs.DS)', 'Emerging Technologies (cs.ET)', 'Formal Languages and Automata Theory (cs.FL)', 'General Literature (cs.GL)', 'Graphics (cs.GR)', 'Computer Science and Game Theory (cs.GT)', 'Human-Computer Interaction (cs.HC)', 'Information Retrieval (cs.IR)', 'Information Theory (cs.IT)', 'Machine Learning (cs.LG)', 'Logic in Computer Science (cs.LO)', 'Multiagent Systems (cs.MA)', 'Multimedia (cs.MM)', 'Mathematical Software (cs.MS)', 'Numerical Analysis (cs.NA)', 'Neural and Evolutionary Computing (cs.NE)', 'Networking and Internet Architecture (cs.NI)', 'Other Computer Science (cs.OH)', 'Operating Systems (cs.OS)', 'Performance (cs.PF)', 'Programming Languages (cs.PL)', 'Robotics (cs.RO)', 'Symbolic Computation (cs.SC)', 'Sound (cs.SD)', 'Software Engineering (cs.SE)', 'Social and Information Networks (cs.SI)', 'Systems and Control (cs.SY)', 'Econometrics (econ.EM)', 'General Economics (econ.GN)', 'Theoretical Economics (econ.TH)', 'Audio and Speech Processing (eess.AS)', 'Image and Video Processing (eess.IV)', 'Signal Processing (eess.SP)', 'Systems and Control (eess.SY)', 'Commutative Algebra (math.AC)', 'Algebraic Geometry (math.AG)', 'Analysis of PDEs (math.AP)', 'Algebraic Topology (math.AT)', 'Classical Analysis and ODEs (math.CA)', 'Combinatorics (math.CO)', 'Category Theory (math.CT)', 'Complex Variables (math.CV)', 'Differential Geometry (math.DG)', 'Dynamical Systems (math.DS)', 'Functional Analysis (math.FA)', 'General Mathematics (math.GM)', 'General Topology (math.GN)', 'Group Theory (math.GR)', 'Geometric Topology (math.GT)', 'History and Overview (math.HO)', 'Information Theory (math.IT)', 'K-Theory and Homology (math.KT)', 'Logic (math.LO)', 'Metric Geometry (math.MG)', 'Mathematical Physics (math.MP)', 'Numerical Analysis (math.NA)', 'Number Theory (math.NT)', 'Operator Algebras (math.OA)', 'Optimization and Control (math.OC)', 'Probability (math.PR)', 'Quantum Algebra (math.QA)', 'Rings and Algebras (math.RA)', 'Representation Theory (math.RT)', 'Symplectic Geometry (math.SG)', 'Spectral Theory (math.SP)', 'Statistics Theory (math.ST)', 'Cosmology and Nongalactic Astrophysics (astro-ph.CO)', 'Earth and Planetary Astrophysics (astro-ph.EP)', 'Astrophysics of Galaxies (astro-ph.GA)', 'High Energy Astrophysical Phenomena (astro-ph.HE)', 'Instrumentation and Methods for Astrophysics (astro-ph.IM)', 'Solar and Stellar Astrophysics (astro-ph.SR)', 'Disordered Systems and Neural Networks (cond-mat.dis-nn)', 'Mesoscale and Nanoscale Physics (cond-mat.mes-hall)', 'Materials Science (cond-mat.mtrl-sci)', 'Other Condensed Matter (cond-mat.other)', 'Quantum Gases (cond-mat.quant-gas)', 'Soft Condensed Matter (cond-mat.soft)', 'Statistical Mechanics (cond-mat.stat-mech)', 'Strongly Correlated Electrons (cond-mat.str-el)', 'Superconductivity (cond-mat.supr-con)', 'General Relativity and Quantum Cosmology (gr-qc)', 'High Energy Physics - Experiment (hep-ex)', 'High Energy Physics - Lattice (hep-lat)', 'High Energy Physics - Phenomenology (hep-ph)', 'High Energy Physics - Theory (hep-th)', 'Mathematical Physics (math-ph)', 'Adaptation and Self-Organizing Systems (nlin.AO)', 'Chaotic Dynamics (nlin.CD)', 'Cellular Automata and Lattice Gases (nlin.CG)', 'Pattern Formation and Solitons (nlin.PS)', 'Exactly Solvable and Integrable Systems (nlin.SI)', 'Nuclear Experiment (nucl-ex)', 'Nuclear Theory (nucl-th)', 'Accelerator Physics (physics.acc-ph)', 'Atmospheric and Oceanic Physics (physics.ao-ph)', 'Applied Physics (physics.app-ph)', 'Atomic and Molecular Clusters (physics.atm-clus)', 'Atomic Physics (physics.atom-ph)', 'Biological Physics (physics.bio-ph)', 'Chemical Physics (physics.chem-ph)', 'Classical Physics (physics.class-ph)', 'Computational Physics (physics.comp-ph)', 'Data Analysis, Statistics and Probability (physics.data-an)', 'Physics Education (physics.ed-ph)', 'Fluid Dynamics (physics.flu-dyn)', 'General Physics (physics.gen-ph)', 'Geophysics (physics.geo-ph)', 'History and Philosophy of Physics (physics.hist-ph)', 'Instrumentation and Detectors (physics.ins-det)', 'Medical Physics (physics.med-ph)', 'Optics (physics.optics)', 'Plasma Physics (physics.plasm-ph)', 'Popular Physics (physics.pop-ph)', 'Physics and Society (physics.soc-ph)', 'Space Physics (physics.space-ph)', 'Quantum Physics (quant-ph)', 'Biomolecules (q-bio.BM)', 'Cell Behavior (q-bio.CB)', 'Genomics (q-bio.GN)', 'Molecular Networks (q-bio.MN)', 'Neurons and Cognition (q-bio.NC)', 'Other Quantitative Biology (q-bio.OT)', 'Populations and Evolution (q-bio.PE)', 'Quantitative Methods (q-bio.QM)', 'Subcellular Processes (q-bio.SC)', 'Tissues and Organs (q-bio.TO)', 'Computational Finance (q-fin.CP)', 'Economics (q-fin.EC)', 'General Finance (q-fin.GN)', 'Mathematical Finance (q-fin.MF)', 'Portfolio Management (q-fin.PM)', 'Pricing of Securities (q-fin.PR)', 'Risk Management (q-fin.RM)', 'Statistical Finance (q-fin.ST)', 'Trading and Market Microstructure (q-fin.TR)', 'Applications (stat.AP)', 'Computation (stat.CO)', 'Methodology (stat.ME)', 'Machine Learning (stat.ML)', 'Other Statistics (stat.OT)', 'Statistics Theory (stat.TH)'] | |
| from transformers import AutoTokenizer, AutoModelForSequenceClassification | |
| import torch | |
| model_name = 'distilbert-base-cased' | |
| device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') | |
| def LoadModel(): | |
| revived = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=len(terms_list)).to(device) | |
| return revived | |
| revived = LoadModel() | |
| revived.load_state_dict(torch.load(model_path, map_location=torch.device(device))) | |
| from torch import nn | |
| from transformers import Trainer, TrainingArguments | |
| training_args = TrainingArguments(output_dir="test_trainer", evaluation_strategy='epoch', | |
| per_device_train_batch_size=4, per_device_eval_batch_size=4) | |
| trainer = Trainer( | |
| model=revived, | |
| args=training_args | |
| ) | |
| tokenizer = AutoTokenizer.from_pretrained(model_name) | |
| intoken = tokenizer(text, padding="max_length", truncation=False) | |
| logits = trainer.predict([intoken]) | |
| predicted_probs = torch.softmax(torch.Tensor(logits.predictions), dim=1).tolist()[0] | |
| predictions = [(x, _ * 100) for _, x in sorted(zip(predicted_probs, terms_list), reverse=True)] | |
| def Top95(predictions): | |
| top95 = list() | |
| sum_prob = 0 | |
| for pred in predictions: | |
| top95.append(pred) | |
| sum_prob += pred[1] | |
| if (sum_prob > 95): | |
| if (len(top95) > 1): | |
| top95 = top95[:-1] | |
| break | |
| return top95 | |
| def PrintResults(top95): | |
| for pred in top95: | |
| st.markdown('{}\t({} %)\n'.format( pred[0], round(pred[1],2) )) | |
| st.markdown("### Top-95% predicted arxiv.org taxonomy:") | |
| PrintResults(Top95(predictions)) | |
| st.markdown("### Нет войне!") | |
| st.markdown("<img width=400px src='https://media.istockphoto.com/photos/flag-of-ukraine-picture-id1369393690?k=20&m=1369393690&s=612x612&w=0&h=bJVj74TIjODPSUTZ5DYESDiBXRo--QyW9LcD4OkLmN0='>", unsafe_allow_html=True) | |