Spaces:
Runtime error
Runtime error
File size: 413 Bytes
f17eb5d 3ae2dd1 f47dd3c 7f6b51d f17eb5d afd14d3 7f6b51d f17eb5d 7f6b51d f17eb5d 7f6b51d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import gradio as gr
import spaces
from transformers import pipeline
import os
import torch
print(torch.cuda.is_available())
summarizer = pipeline("summarization", model="ShynBui/Bartpho_spelling_correction", device=0)
@spaces.GPU
def generate(prompt):
return summarizer(prompt, max_new_tokens = 512)[0]['summary_text']
gr.Interface(
fn=generate,
inputs=gr.Text(),
outputs=gr.Text(),
).launch() |