ManthaBhuvana commited on
Commit
580976d
·
verified ·
1 Parent(s): a8acd31

Update mcq_generator.py

Browse files
Files changed (1) hide show
  1. mcq_generator.py +11 -10
mcq_generator.py CHANGED
@@ -9,19 +9,20 @@ Original file is located at
9
  #MCQ Generation Using T5
10
  """
11
 
12
- # Improved MCQ Generator using T5 Model
13
- import torch
14
- from transformers import T5ForConditionalGeneration, T5Tokenizer, AutoModelForSeq2SeqLM, AutoTokenizer,pipeline
15
  import nltk
16
  import random
17
- from nltk.tokenize import sent_tokenize, word_tokenize
18
- from nltk.corpus import stopwords
19
- from nltk.tag import pos_tag
20
- from sklearn.feature_extraction.text import TfidfVectorizer
21
- from sklearn.metrics.pairwise import cosine_similarity
22
- import numpy as np
23
  import re
24
- import string
 
 
 
 
 
 
 
 
 
 
25
 
26
  # Download required NLTK packages
27
  nltk.download('punkt')
 
9
  #MCQ Generation Using T5
10
  """
11
 
 
 
 
12
  import nltk
13
  import random
 
 
 
 
 
 
14
  import re
15
+ import torch
16
+ import spacy
17
+ import pandas as pd
18
+ import numpy as np
19
+ from sklearn.feature_extraction.text import TfidfVectorizer
20
+ from transformers import pipeline, T5ForConditionalGeneration, T5Tokenizer, AutoModelForQuestionAnswering, AutoTokenizer
21
+ from nltk.corpus import wordnet as wn
22
+ from nltk.corpus import stopwords
23
+ from nltk import pos_tag, word_tokenize
24
+ from sentence_transformers import SentenceTransformer, util
25
+ from rouge import Rouge
26
 
27
  # Download required NLTK packages
28
  nltk.download('punkt')