/SpaceQQuiz

Primary LanguagePythonApache License 2.0Apache-2.0

SpaceQQuiz

SpaceQQuiz is a system to generate quizzes, a common resource to evaluate training sessions, out of quality procedure documents in the Space domain. Our system leverages state of the art auto-regressive models like T5 and BART to generate questions, and a RoBERTa model to extract answer for the questions, thus verifying their suitability.

Requirements:

Installation:

Create a new conda environment:

conda create -n spaceqquiz python=3.9
conda activate spaceqquiz
cd SpaceQQuiz/
pip install -r requirements.txt

Execution

streamlit run run_question_generation.py -- --question_generation_endpoint=$QUESTION_GENERATION_ENDPOINT

Question Generation Module

Create question-generation conda environment and install required libraries (for GPU use, check CUDA version):

conda create -n question-generation python=3.7
conda activate question-generation
cd SpaceQQuiz/question-generation
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
pip install -r requirements.txt

To run the question-generation module (You need to download the question-generation-squad-bart-large and question-generation-squad-t5-large):

python src/app.py

By default the endpoints will be: