Hugging Face's logo

Spaces:
TharvinPrakash
/
Multi-model-Chatbot
Sleeping

App Files Community
TharvinPrakash commited on
Commit
8fda1c7
·
verified ·
1 Parent(s): 60ad903

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import torch
6
  @st.cache_resource
7
  def load_re_punctuate_model():
8
  tokenizer = AutoTokenizer.from_pretrained("SJ-Ray/Re-Punctuate")
9
- model = AutoModelForSeq2SeqLM.from_pretrained("SJ-Ray/Re-Punctuate")
10
  return tokenizer, model
11
 
12
  # Load Hugging Face tokenizer and model for headline generation
 
6
  @st.cache_resource
7
  def load_re_punctuate_model():
8
  tokenizer = AutoTokenizer.from_pretrained("SJ-Ray/Re-Punctuate")
9
+ model = AutoModelForSeq2SeqLM.from_pretrained("SJ-Ray/Re-Punctuate",from_tf=True)
10
  return tokenizer, model
11
 
12
  # Load Hugging Face tokenizer and model for headline generation