Curative commited on
Commit
f506df8
·
verified ·
1 Parent(s): d2f7e2a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # Load the summarization model
5
- summarizer = pipeline("summarization", model="AventIQ-AI/t5-text-summarizer")
6
 
7
  # Define the summarization function
8
  def summarize_text(input_text):
 
2
  from transformers import pipeline
3
 
4
  # Load the summarization model
5
+ summarizer = pipeline("summarization", model="Curative/t5-summarizer-cnn")
6
 
7
  # Define the summarization function
8
  def summarize_text(input_text):