kazimsayed commited on
Commit
46e2fd5
·
1 Parent(s): 3ca4173

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -1,8 +1,6 @@
1
  import random
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import gradio as gr
4
- from gramformer import Gramformer
5
- import spacy
6
 
7
  tokenizer = AutoTokenizer.from_pretrained("docto/Docto-Bot")
8
  model = AutoModelForCausalLM.from_pretrained("docto/Docto-Bot")
@@ -32,8 +30,8 @@ def get_reply(userinput):
32
  try:
33
  result = tokenizer.decode(random.choice(output_sequences))
34
  result = result[result.index("Answer: "):result.index(special_token)]
35
- corrected_sentence = gf.correct(result[8:], max_candidates=1)
36
- return corrected_sentence
37
  except:
38
  return "Sorry! I don\'t Know"
39
 
 
1
  import random
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import gradio as gr
 
 
4
 
5
  tokenizer = AutoTokenizer.from_pretrained("docto/Docto-Bot")
6
  model = AutoModelForCausalLM.from_pretrained("docto/Docto-Bot")
 
30
  try:
31
  result = tokenizer.decode(random.choice(output_sequences))
32
  result = result[result.index("Answer: "):result.index(special_token)]
33
+ return (result[8:])
34
+
35
  except:
36
  return "Sorry! I don\'t Know"
37