ZeeAI1 commited on
Commit
19cfcf0
·
verified ·
1 Parent(s): be9edff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  # Load model and tokenizer from Hugging Face
6
  @st.cache_resource
7
  def load_model():
8
- model_name = "vennify/t5-base-grammar-correction"
9
  tokenizer = T5Tokenizer.from_pretrained(model_name)
10
  model = T5ForConditionalGeneration.from_pretrained(model_name)
11
  return tokenizer, model
 
5
  # Load model and tokenizer from Hugging Face
6
  @st.cache_resource
7
  def load_model():
8
+ model_name = model_name = "pszemraj/flan-t5-large-grammar-synthesis"
9
  tokenizer = T5Tokenizer.from_pretrained(model_name)
10
  model = T5ForConditionalGeneration.from_pretrained(model_name)
11
  return tokenizer, model