smfaiz commited on
Commit
cd8d72a
·
verified ·
1 Parent(s): e4ff6be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,8 +64,8 @@ def extract_text_from_html(url):
64
  except Exception as e:
65
  return f"Error extracting text: {str(e)}"
66
 
 
67
  tokenizer = AutoTokenizer.from_pretrained("google/pegasus-x-large")
68
- model = PegasusModel.from_pretrained("google/pegasus-x-large")
69
 
70
  def summarize_article(article_text):
71
  """Summarize a given article's text."""
 
64
  except Exception as e:
65
  return f"Error extracting text: {str(e)}"
66
 
67
+ model = PegasusXForConditionalGeneration.from_pretrained("google/pegasus-x-base")
68
  tokenizer = AutoTokenizer.from_pretrained("google/pegasus-x-large")
 
69
 
70
  def summarize_article(article_text):
71
  """Summarize a given article's text."""