Spaces:

Karzan
/

bart-ckb-summarization

Runtime error

Karzan commited on Jul 20, 2024

Commit

b052da8

verified ·

1 Parent(s): 94c2e2b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 import spaces
 from transformers import pipeline
-model_name = 'Karzan/bart-summarization'
 summarizer = pipeline("summarization", model=model_name,device='cuda')
 @spaces.GPU(duration=120)

 import spaces
 from transformers import pipeline
+model_name = 'Karzan/bart-ckb-summarization'
 summarizer = pipeline("summarization", model=model_name,device='cuda')
 @spaces.GPU(duration=120)