Update app.py
Browse files
app.py
CHANGED
@@ -8,37 +8,32 @@ text_generator = pipeline('text-generation', model='gpt2')
|
|
8 |
|
9 |
pos_tags = ['ADJ', 'ADP', 'ADV', 'AUX', 'CONJ', 'DET', 'INTJ', 'NOUN', 'NUM', 'PART', 'PRON', 'PROPN', 'PUNCT', 'SCONJ', 'SYM', 'VERB', 'X']
|
10 |
|
11 |
-
sentence_state = {'sentence': '', 'tagged_words': []}
|
12 |
-
|
13 |
def generate_sentence():
|
14 |
result = text_generator('')[0]
|
15 |
sentence = result['generated_text']
|
16 |
tagged_words = analyze_sentence(sentence)
|
17 |
-
sentence_state['sentence'] = sentence
|
18 |
-
sentence_state['tagged_words'] = tagged_words
|
19 |
return sentence, [word for word, _ in tagged_words]
|
20 |
|
21 |
def analyze_sentence(sentence):
|
22 |
doc = nlp(sentence)
|
23 |
return [(token.text, token.pos_) for token in doc]
|
24 |
|
25 |
-
def check_answer(*args):
|
26 |
-
|
|
|
27 |
user_answer = list(args)
|
28 |
if user_answer == correct_answer:
|
29 |
return 'Correcto!'
|
30 |
else:
|
31 |
return 'Incorrecto. La respuesta correcta es: ' + str(correct_answer)
|
32 |
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
-
gr.outputs.Textbox(label='Result')])
|
44 |
-
iface.launch()
|
|
|
8 |
|
9 |
pos_tags = ['ADJ', 'ADP', 'ADV', 'AUX', 'CONJ', 'DET', 'INTJ', 'NOUN', 'NUM', 'PART', 'PRON', 'PROPN', 'PUNCT', 'SCONJ', 'SYM', 'VERB', 'X']
|
10 |
|
|
|
|
|
11 |
def generate_sentence():
|
12 |
result = text_generator('')[0]
|
13 |
sentence = result['generated_text']
|
14 |
tagged_words = analyze_sentence(sentence)
|
|
|
|
|
15 |
return sentence, [word for word, _ in tagged_words]
|
16 |
|
17 |
def analyze_sentence(sentence):
|
18 |
doc = nlp(sentence)
|
19 |
return [(token.text, token.pos_) for token in doc]
|
20 |
|
21 |
+
def check_answer(sentence, *args):
|
22 |
+
tagged_words = analyze_sentence(sentence)
|
23 |
+
correct_answer = [tag for word, tag in tagged_words]
|
24 |
user_answer = list(args)
|
25 |
if user_answer == correct_answer:
|
26 |
return 'Correcto!'
|
27 |
else:
|
28 |
return 'Incorrecto. La respuesta correcta es: ' + str(correct_answer)
|
29 |
|
30 |
+
generate_iface = gr.Interface(fn=generate_sentence,
|
31 |
+
inputs='number',
|
32 |
+
outputs=['textbox', 'textbox'])
|
33 |
+
|
34 |
+
check_iface = gr.Interface(fn=check_answer,
|
35 |
+
inputs=['textbox'] + [gr.inputs.Dropdown(choices=pos_tags) for _ in range(10)],
|
36 |
+
outputs='textbox')
|
37 |
+
|
38 |
+
generate_iface.launch()
|
39 |
+
check_iface.launch()
|
|
|
|