asd

app.py

@@ -224,6 +224,7 @@ def classify_text(text):
     main_score = probabilities[0][predicted_class].item()
     return main_label, main_score
 
+@spaces.GPU
 def correct_grammar(text):
     inputs = grammar_tokenizer(f'Fix grammatical errors in this sentence: {text}', return_tensors="pt").input_ids.to(device)
     outputs = grammar_model.generate(inputs, max_length=256)