Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|