mynkchaudhry commited on
Commit
cb4dd2c
·
verified ·
1 Parent(s): e2aedb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
  import gradio as gr
3
 
4
  # Load model and tokenizer
5
- tokenizer = AutoTokenizer.from_pretrained("mynkchaudhry/Summarization-Pro")
6
- model = AutoModelForSeq2SeqLM.from_pretrained("mynkchaudhry/Summarization-Pro")
7
 
8
  def summarize(text):
9
  inputs = tokenizer(text, return_tensors="pt", max_length=1024, truncation=True)
 
2
  import gradio as gr
3
 
4
  # Load model and tokenizer
5
+ tokenizer = AutoTokenizer.from_pretrained("mynkchaudhry/Summarization-Pro",force_download=True,cache_dir=None)
6
+ model = AutoModelForSeq2SeqLM.from_pretrained("mynkchaudhry/Summarization-Pro",force_download=True,cache_dir=None)
7
 
8
  def summarize(text):
9
  inputs = tokenizer(text, return_tensors="pt", max_length=1024, truncation=True)