arfat-xyz commited on
Commit
497d248
1 Parent(s): d4a2f4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
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)