ColeGuion commited on
Commit
b77eb6c
·
verified ·
1 Parent(s): f27161f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -15
app.py CHANGED
@@ -10,23 +10,11 @@ client = InferenceClient("vennify/t5-base-grammar-correction")
10
  #gr.load("models/vennify/t5-base-grammar-correction").launch()
11
 
12
  # Load the model and tokenizer
13
- model_name = "vennify/t5-base-grammar-correction"
14
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
15
- tokenizer = AutoTokenizer.from_pretrained(model_name)
16
 
17
 
18
- # Define the function to generate corrected text
19
- def correct_text(text, max_length, num_beams, temperature):
20
- inputs = tokenizer.encode(text, return_tensors="pt")
21
- outputs = model.generate(
22
- inputs,
23
- max_length=max_length,
24
- num_beams=num_beams,
25
- temperature=temperature,
26
- early_stopping=True
27
- )
28
- corrected_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
29
- return corrected_text
30
 
31
 
32
  def respond(
 
10
  #gr.load("models/vennify/t5-base-grammar-correction").launch()
11
 
12
  # Load the model and tokenizer
13
+ #model_name = "vennify/t5-base-grammar-correction"
14
+ #model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
15
+ #tokenizer = AutoTokenizer.from_pretrained(model_name)
16
 
17
 
 
 
 
 
 
 
 
 
 
 
 
 
18
 
19
 
20
  def respond(