Mr-Vicky-01 commited on
Commit
decc31b
1 Parent(s): e8233f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
  import gradio as gr
3
 
4
- checkpoint = "Mr-Vicky-01/conversational_sumarization"
5
 
6
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
7
  model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
 
1
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
  import gradio as gr
3
 
4
+ checkpoint = "Mr-Vicky-01/Bart-Finetuned-conversational-summarization"
5
 
6
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
7
  model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)