Spaces:
Sleeping
Sleeping
Commit
•
84710d4
1
Parent(s):
ef98ab3
Update app.py
Browse files
app.py
CHANGED
@@ -4,11 +4,10 @@ import gradio as gr
|
|
4 |
from transformers import T5ForConditionalGeneration, T5Tokenizer
|
5 |
#question_model = T5ForConditionalGeneration.from_pretrained('ramsrigouthamg/t5_squad_v1')
|
6 |
#question_tokenizer = T5Tokenizer.from_pretrained('t5-base')
|
7 |
-
|
|
|
8 |
|
9 |
def get_question(context, answer):
|
10 |
-
question_model = T5ForConditionalGeneration.from_pretrained('ramsrigouthamg/t5_squad_v1')
|
11 |
-
question_tokenizer = T5Tokenizer.from_pretrained('t5-small')
|
12 |
text = "context: {} answer: {}".format(context, answer)
|
13 |
#max_len = 512
|
14 |
#encoding = question_tokenizer.encode_plus(text, max_length=max_len, padding='max_length', truncation=True, return_tensors="pt")
|
|
|
4 |
from transformers import T5ForConditionalGeneration, T5Tokenizer
|
5 |
#question_model = T5ForConditionalGeneration.from_pretrained('ramsrigouthamg/t5_squad_v1')
|
6 |
#question_tokenizer = T5Tokenizer.from_pretrained('t5-base')
|
7 |
+
question_model = T5ForConditionalGeneration.from_pretrained('ramsrigouthamg/t5_squad_v1')
|
8 |
+
question_tokenizer = T5Tokenizer.from_pretrained('t5-small')
|
9 |
|
10 |
def get_question(context, answer):
|
|
|
|
|
11 |
text = "context: {} answer: {}".format(context, answer)
|
12 |
#max_len = 512
|
13 |
#encoding = question_tokenizer.encode_plus(text, max_length=max_len, padding='max_length', truncation=True, return_tensors="pt")
|