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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -8,8 +8,10 @@ from gradio.mix import Parallel, Series
8
 
9
  desc = "Summarize your text! (audio transcription available soon)"
10
 
 
 
11
  def summarize(text):
12
- summ = gr.Interface.load('huggingface/google/pegasus-large')
13
  summary = summ(text)
14
  return summary
15
  iface = gr.Interface(fn=summarize,
 
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,