Spaces:

astro21
/

test_2

Sleeping

astro21 commited on Oct 10, 2023

Commit

950ec7b

1 Parent(s): e02cc8f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ summarizer = pipeline("summarization", model="astro21/bart-cls")
 def summarize_text(input_text):
     # Use the summarization pipeline to generate a summary
-    summarized_text = summarizer(input_text, max_length=128, min_length=64, do_sample=False, wait_for_model=True)
     return summarized_text[0]['summary_text']
 demo = gr.Interface(fn= summarize_text , inputs="text", outputs="text")

 def summarize_text(input_text):
     # Use the summarization pipeline to generate a summary
+    summarized_text = summarizer(input_text, max_length=128, min_length=64, do_sample=False)
     return summarized_text[0]['summary_text']
 demo = gr.Interface(fn= summarize_text , inputs="text", outputs="text")