QAway-to commited on
Commit
a5e0a96
·
1 Parent(s): e8b5e7a

google/flan-t5-small . app.py v1.9

Browse files
Files changed (1) hide show
  1. core/interviewer.py +2 -2
core/interviewer.py CHANGED
@@ -4,10 +4,10 @@ import difflib
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
5
 
6
  # Выбери одну
7
- QG_MODEL = "mrm8488/t5-small-finetuned-question-generation-ap"
8
- # QG_MODEL = "google/flan-t5-small"
9
  # QG_MODEL = "iarfmoose/t5-base-question-generator"
10
 
 
11
  tokenizer = AutoTokenizer.from_pretrained(QG_MODEL)
12
  model = AutoModelForSeq2SeqLM.from_pretrained(QG_MODEL)
13
 
 
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
5
 
6
  # Выбери одну
7
+ QG_MODEL = "google/flan-t5-small"
 
8
  # QG_MODEL = "iarfmoose/t5-base-question-generator"
9
 
10
+
11
  tokenizer = AutoTokenizer.from_pretrained(QG_MODEL)
12
  model = AutoModelForSeq2SeqLM.from_pretrained(QG_MODEL)
13