SamuelMiller commited on
Commit
bf26be5
·
1 Parent(s): 1244cc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -9,11 +9,13 @@ desc = "Summarize your text! (audio transcription available soon)"
9
 
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,
 
9
 
10
 
11
  qa_model = 'huggingface/SamuelMiller/qa_squad'
12
+ my_model_1 = 'huggingface/SamuelMiller/lil_sumsum'
13
+ my_model_2 = 'huggingface/SamuelMiller/lil_sum_sum'
14
+ my_model_3 = 'huggingface/SamuelMiller/sum_sum'
15
  better_model = 'huggingface/google/pegasus-large'
16
 
17
  def summarize(text):
18
+ summ = gr.Interface.load(my_model_3)
19
  summary = summ(text)
20
  return summary
21
  iface = gr.Interface(fn=summarize,