Update app.py
Browse files
app.py
CHANGED
@@ -24,6 +24,12 @@ from sklearn.metrics.pairwise import cosine_similarity
|
|
24 |
import nltk
|
25 |
nltk.download('omw-1.4')
|
26 |
import gradio as gr
|
|
|
|
|
|
|
|
|
|
|
|
|
27 |
question_model = T5ForConditionalGeneration.from_pretrained('ramsrigouthamg/t5_squad_v1')
|
28 |
question_tokenizer = T5Tokenizer.from_pretrained('ramsrigouthamg/t5_squad_v1')
|
29 |
question_model = question_model.to(device)
|
|
|
24 |
import nltk
|
25 |
nltk.download('omw-1.4')
|
26 |
import gradio as gr
|
27 |
+
|
28 |
+
|
29 |
+
from transformers import T5ForConditionalGeneration,T5Tokenizer
|
30 |
+
summary_model = T5ForConditionalGeneration.from_pretrained('t5-base')
|
31 |
+
summary_tokenizer = T5Tokenizer.from_pretrained('t5-base')
|
32 |
+
|
33 |
question_model = T5ForConditionalGeneration.from_pretrained('ramsrigouthamg/t5_squad_v1')
|
34 |
question_tokenizer = T5Tokenizer.from_pretrained('ramsrigouthamg/t5_squad_v1')
|
35 |
question_model = question_model.to(device)
|