Spaces:
Runtime error
Runtime error
first
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
|
|
5 |
pipe = pipeline("summarization", model="ShynBui/Bartpho_spelling_correction")
|
6 |
|
7 |
def greet(text):
|
8 |
-
return pipe(text, max_new_tokens= 512)['summary_text']
|
9 |
|
10 |
demo = gr.Interface(fn=greet, inputs=gr.Text(), outputs=gr.Text())
|
11 |
demo.launch()
|
|
|
5 |
pipe = pipeline("summarization", model="ShynBui/Bartpho_spelling_correction")
|
6 |
|
7 |
def greet(text):
|
8 |
+
return pipe(text, max_new_tokens= 512)[0]['summary_text']
|
9 |
|
10 |
demo = gr.Interface(fn=greet, inputs=gr.Text(), outputs=gr.Text())
|
11 |
demo.launch()
|