Spaces:
Runtime error
Runtime error
models
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ examples = [
|
|
19 |
Q: A juggler can juggle 16 balls. Half of the balls are golf balls, and half of the golf balls are blue. How many blue golf balls are there?"""]
|
20 |
]
|
21 |
|
22 |
-
title = "Upword.
|
23 |
|
24 |
token = os.environ["token"]
|
25 |
|
@@ -27,7 +27,7 @@ urls = {
|
|
27 |
'flan-t5': os.environ["url"],
|
28 |
# 'flan-t5': "https://api-inference.huggingface.co/models/philschmid/flan-t5-xxl-sharded-fp16",
|
29 |
'bart-large-cnn': "https://api-inference.huggingface.co/models/facebook/bart-large-cnn",
|
30 |
-
'pegasus': "https://api-inference.huggingface.co/models/google/pegasus-xsum"
|
31 |
}
|
32 |
|
33 |
|
@@ -47,8 +47,8 @@ def inference(text, min_length, max_length):
|
|
47 |
|
48 |
output_flan = responses['flan-t5'].json()[0]['generated_text']
|
49 |
output_bart = responses['bart-large-cnn'].json()[0]['summary_text']
|
50 |
-
output_gpt = responses['pegasus'].json()[0]['summary_text']
|
51 |
-
return [output_flan, output_bart
|
52 |
|
53 |
|
54 |
io = gr.Interface(
|
|
|
19 |
Q: A juggler can juggle 16 balls. Half of the balls are golf balls, and half of the golf balls are blue. How many blue golf balls are there?"""]
|
20 |
]
|
21 |
|
22 |
+
title = "Upword. Models Playground."
|
23 |
|
24 |
token = os.environ["token"]
|
25 |
|
|
|
27 |
'flan-t5': os.environ["url"],
|
28 |
# 'flan-t5': "https://api-inference.huggingface.co/models/philschmid/flan-t5-xxl-sharded-fp16",
|
29 |
'bart-large-cnn': "https://api-inference.huggingface.co/models/facebook/bart-large-cnn",
|
30 |
+
# 'pegasus': "https://api-inference.huggingface.co/models/google/pegasus-xsum"
|
31 |
}
|
32 |
|
33 |
|
|
|
47 |
|
48 |
output_flan = responses['flan-t5'].json()[0]['generated_text']
|
49 |
output_bart = responses['bart-large-cnn'].json()[0]['summary_text']
|
50 |
+
# output_gpt = responses['pegasus'].json()[0]['summary_text']
|
51 |
+
return [output_flan, output_bart]
|
52 |
|
53 |
|
54 |
io = gr.Interface(
|