tanishkaguha commited on
Commit
adef624
·
verified ·
1 Parent(s): 2919b23

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -10,7 +10,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
10
  # Check if GPU is available
11
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
  model = model.to(device)
13
-
14
  # Grammar correction function
15
  def correct_grammar(text):
16
  input_text = "gec: " + text
 
10
  # Check if GPU is available
11
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
  model = model.to(device)
13
+
14
  # Grammar correction function
15
  def correct_grammar(text):
16
  input_text = "gec: " + text