Commit
·
46e2fd5
1
Parent(s):
3ca4173
Update app.py
Browse files
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 |
-
|
36 |
-
|
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 |
|