Commit
·
1ac728b
1
Parent(s):
6ff692c
Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
import random
|
2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
|
3 |
|
4 |
tokenizer = AutoTokenizer.from_pretrained("docto/Docto-Bot")
|
5 |
model = AutoModelForCausalLM.from_pretrained("docto/Docto-Bot")
|
@@ -30,4 +31,7 @@ def get_reply(userinput):
|
|
30 |
return(result[8:])
|
31 |
except:
|
32 |
return "Sorry! I don\'t Know"
|
|
|
|
|
|
|
33 |
|
|
|
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")
|
|
|
31 |
return(result[8:])
|
32 |
except:
|
33 |
return "Sorry! I don\'t Know"
|
34 |
+
|
35 |
+
iface = gr.Interface(fn=get_reply, inputs=["text"], outputs=["textbox"]).launch()
|
36 |
+
|
37 |
|