jcockhren commited on
Commit
3bc75cc
·
verified ·
1 Parent(s): 31f212e

use bart large

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ PROVIDERMODEL = {
12
 
13
  SUMMARYMODEL = {
14
  'openai': 'openai/whisper-medium.en',
15
- 'facebook': 'facebook/bart-medium-en',
16
  }
17
 
18
  def query(filename, provider='openai'):
@@ -27,7 +27,7 @@ def query(filename, provider='openai'):
27
  return response.content
28
 
29
 
30
- def summarize(text, provider='openai'):
31
  data = {
32
  'inputs': text,
33
  'parameters': {
 
12
 
13
  SUMMARYMODEL = {
14
  'openai': 'openai/whisper-medium.en',
15
+ 'facebook': 'facebook/bart-large',
16
  }
17
 
18
  def query(filename, provider='openai'):
 
27
  return response.content
28
 
29
 
30
+ def summarize(text, provider='facebook'):
31
  data = {
32
  'inputs': text,
33
  'parameters': {