AbdurRehman313 commited on
Commit
04f892b
·
verified ·
1 Parent(s): 20ffdcc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -22,14 +22,14 @@ import streamlit as st
22
  import requests
23
  import torch
24
  from transformers import pipeline
25
- from transformers import T5ForConditionalGeneration, T5Tokenizer
26
 
27
  # Replace with your Hugging Face model repository path
28
  model_repo_path = 'AbdurRehman313/BART_samsum'
29
 
30
  # Load the model and tokenizer
31
- model = T5ForConditionalGeneration.from_pretrained(model_repo_path)
32
- tokenizer = T5Tokenizer.from_pretrained(model_repo_path)
33
 
34
  # Initialize the summarization pipeline
35
  summarizer = pipeline('summarization', model=model,tokenizer=tokenizer)
 
22
  import requests
23
  import torch
24
  from transformers import pipeline
25
+ from transformers import BartTokenizer, BartForConditionalGeneration
26
 
27
  # Replace with your Hugging Face model repository path
28
  model_repo_path = 'AbdurRehman313/BART_samsum'
29
 
30
  # Load the model and tokenizer
31
+ model = BartForConditionalGeneration.from_pretrained(model_repo_path)
32
+ tokenizer = BartTokenizer.from_pretrained(model_repo_path)
33
 
34
  # Initialize the summarization pipeline
35
  summarizer = pipeline('summarization', model=model,tokenizer=tokenizer)