SamuelMiller commited on
Commit
9e789ed
·
1 Parent(s): 037452a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -8,10 +8,12 @@ from gradio.mix import Parallel, Series
8
 
9
  desc = "Summarize your text! (audio transcription available soon)"
10
 
 
11
  my_model = 'huggingface/SamuelMiller/lil_sumsum'
12
  better_model = 'huggingface/google/pegasus-large'
 
13
  def summarize(text):
14
- summ = gr.Interface.load(my_model)
15
  summary = summ(text)
16
  return summary
17
  iface = gr.Interface(fn=summarize,
 
8
 
9
  desc = "Summarize your text! (audio transcription available soon)"
10
 
11
+ qa_model = 'huggingface/SamuelMiller/qa_squad'
12
  my_model = 'huggingface/SamuelMiller/lil_sumsum'
13
  better_model = 'huggingface/google/pegasus-large'
14
+
15
  def summarize(text):
16
+ summ = gr.Interface.load(qa_model)
17
  summary = summ(text)
18
  return summary
19
  iface = gr.Interface(fn=summarize,