Welcome to the Deep Paper Predictor GitHub repository! This repository contains code for a web application that assists researchers in finding relevant research papers and predicting the subject area of a paper based on its abstract.
The Deep Paper Predictor is a web application developed using Streamlit, TensorFlow, PyTorch, and Sentence Transformers. It consists of two main functionalities:
- Paper Recommendation: Given the title of a research paper, the application recommends similar papers based on cosine similarity scores calculated using sentence embeddings.
- Subject Area Prediction: Given the abstract of a research paper, the application predicts the subject area using a shallow Multi-Layer Perceptron (MLP) model.
To use the Deep Paper Predictor:
- Clone this repository to your local machine.
- Install the required dependencies listed in the
requirements.txtfile. - Need to save all the models form the Research Paper recommendation and subject area prediction using sentence transformer.ipyb file.
- There will be total 7 files in the models folder so be sure that you have save all the 7 models.
- Run the Streamlit application by executing
streamlit run app.pyin your terminal. - Enter the details of the paper (title and abstract) in the sidebar.
- Click on the "Recommend" button to get recommendations and predicted subject areas.
- TensorFlow
- PyTorch
- Sentence Transformers
- Streamlit
- Googlesearch
All dependencies can be installed via pip using the command pip install -r requirements.txt.
- app.py: Main Streamlit application file containing the user interface and interaction logic.
- models/: Directory containing saved models, embeddings, and configurations.
- utils.py: Utility functions for recommendation and subject area prediction.