Spaces:
Running
on
A10G
Running
on
A10G
osanseviero
commited on
Commit
•
911bd1c
1
Parent(s):
263d9d0
Update app.py
Browse files
app.py
CHANGED
@@ -25,8 +25,8 @@ title = "Flan T5 and Vanilla T5"
|
|
25 |
description = "Demo that compares [T5-large](https://huggingface.co/t5-large) and [Flan-T5-large](https://huggingface.co/ybelkada/flan-t5-large)"
|
26 |
|
27 |
def inference(text):
|
28 |
-
output_flan = pipe_flan(text,
|
29 |
-
output_vanilla = pipe_vanilla(text,
|
30 |
return [output_flan, output_vanilla]
|
31 |
|
32 |
io = gr.Interface(
|
|
|
25 |
description = "Demo that compares [T5-large](https://huggingface.co/t5-large) and [Flan-T5-large](https://huggingface.co/ybelkada/flan-t5-large)"
|
26 |
|
27 |
def inference(text):
|
28 |
+
output_flan = pipe_flan(text, max_length=100)[0]["generated_text"]
|
29 |
+
output_vanilla = pipe_vanilla(text, max_length=100)[0]["generated_text"]
|
30 |
return [output_flan, output_vanilla]
|
31 |
|
32 |
io = gr.Interface(
|