flynn-chen commited on
Commit
c6a7684
·
1 Parent(s): 97ec4dd
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,4 +1,8 @@
1
  import os
 
 
 
 
2
  import gradio as gr
3
  import pandas as pd
4
  from question_generation.pipelines import pipeline
@@ -7,8 +11,6 @@ import docx2txt
7
  qa_list = []
8
  def process_file(Notes):
9
 
10
- os.system("pip install -U transformers==3.0.0")
11
- os.system("python -m nltk.downloader punkt")
12
  nlp = pipeline("question-generation", model="valhalla/t5-small-qg-prepend", qg_format="prepend")
13
 
14
  target_word_doc = Notes.name
 
1
  import os
2
+
3
+ os.system("pip install -U transformers==3.0.0")
4
+ os.system("python -m nltk.downloader punkt")
5
+
6
  import gradio as gr
7
  import pandas as pd
8
  from question_generation.pipelines import pipeline
 
11
  qa_list = []
12
  def process_file(Notes):
13
 
 
 
14
  nlp = pipeline("question-generation", model="valhalla/t5-small-qg-prepend", qg_format="prepend")
15
 
16
  target_word_doc = Notes.name