llmahmad commited on
Commit
bd16136
·
verified ·
1 Parent(s): fd40918

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -1,15 +1,14 @@
1
  import os
2
- os.system('pip install streamlit transformers torch sentencepiece')
3
 
4
  import streamlit as st
5
- from transformers import PegasusTokenizer, PegasusForConditionalGeneration
6
 
7
  # Load the model and tokenizer
8
- model_name = '.' # Path to the current directory where files are located
9
 
10
-
11
- tokenizer = PegasusTokenizer.from_pretrained(model_name)
12
- model = PegasusForConditionalGeneration.from_pretrained(model_name)
13
 
14
  def summarize_text(text):
15
  inputs = tokenizer(text, return_tensors="pt", truncation=True, padding="longest")
 
1
  import os
2
+ os.system('pip install streamlit transformers torch')
3
 
4
  import streamlit as st
5
+ from transformers import BartTokenizer, BartForConditionalGeneration
6
 
7
  # Load the model and tokenizer
8
+ model_name = 'llmahmad/facebook_BART_summary'
9
 
10
+ tokenizer = BartTokenizer.from_pretrained(model_name)
11
+ model = BartForConditionalGeneration.from_pretrained(model_name)
 
12
 
13
  def summarize_text(text):
14
  inputs = tokenizer(text, return_tensors="pt", truncation=True, padding="longest")