Hugging Face's logo

Spaces:
mbalajibalu
/
grammar-space
Running on Zero

App Files Community
Kavin0003 commited on
Commit
28e483a
·
1 Parent(s): d7de9f7

✨ Updated grammar with the vennify model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  import torch
4
 
5
  # Load from Hugging Face model hub
6
- model_path = "mbalajibalu/grammar-model"
7
 
8
  tokenizer = AutoTokenizer.from_pretrained(model_path)
9
  model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
 
3
  import torch
4
 
5
  # Load from Hugging Face model hub
6
+ model_path = "vennify/t5-base-grammar-correction"
7
 
8
  tokenizer = AutoTokenizer.from_pretrained(model_path)
9
  model = AutoModelForSeq2SeqLM.from_pretrained(model_path)