Spaces:
Build error
Build error
use bart large
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ PROVIDERMODEL = {
|
|
| 12 |
|
| 13 |
SUMMARYMODEL = {
|
| 14 |
'openai': 'openai/whisper-medium.en',
|
| 15 |
-
'facebook': 'facebook/bart-
|
| 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='
|
| 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': {
|