Spaces:
Sleeping
Sleeping
Commit
·
48bf45d
1
Parent(s):
6900593
Update app.py
Browse files
app.py
CHANGED
@@ -7,6 +7,8 @@ Created on Thu Sep 21 22:17:43 2023
|
|
7 |
|
8 |
import streamlit as st
|
9 |
import sentencepiece
|
|
|
|
|
10 |
from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration
|
11 |
|
12 |
model=T5ForConditionalGeneration.from_pretrained("Logeswaransr/T5_MineAI_Prototype").to("cpu")
|
@@ -14,6 +16,8 @@ tokenizer=T5Tokenizer.from_pretrained("Logeswaransr/T5_MineAI_Prototype")
|
|
14 |
|
15 |
pipe=pipeline('text2text-generation', model=model, tokenizer=tokenizer)
|
16 |
|
|
|
|
|
17 |
greetings=["Hello! My name is MineAI, A specially trained LLM here to assist you on your Mining Related Queries.","How may I help you?"]
|
18 |
|
19 |
st.set_page_config(page_title='Sample Chatbot', layout='wide')
|
|
|
7 |
|
8 |
import streamlit as st
|
9 |
import sentencepiece
|
10 |
+
from gtts import gTTS
|
11 |
+
from io import BytesIO
|
12 |
from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration
|
13 |
|
14 |
model=T5ForConditionalGeneration.from_pretrained("Logeswaransr/T5_MineAI_Prototype").to("cpu")
|
|
|
16 |
|
17 |
pipe=pipeline('text2text-generation', model=model, tokenizer=tokenizer)
|
18 |
|
19 |
+
|
20 |
+
|
21 |
greetings=["Hello! My name is MineAI, A specially trained LLM here to assist you on your Mining Related Queries.","How may I help you?"]
|
22 |
|
23 |
st.set_page_config(page_title='Sample Chatbot', layout='wide')
|