Spaces:
Sleeping
Sleeping
Commit
•
33c6f10
1
Parent(s):
48bf45d
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ tokenizer=T5Tokenizer.from_pretrained("Logeswaransr/T5_MineAI_Prototype")
|
|
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 |
|
@@ -37,6 +37,10 @@ if st.session_state.messages==[]:
|
|
37 |
for gr in greetings:
|
38 |
with st.chat_message("assistant"):
|
39 |
st.markdown(gr)
|
|
|
|
|
|
|
|
|
40 |
|
41 |
st.session_state.messages.append({
|
42 |
'role':'assistant',
|
@@ -57,6 +61,10 @@ if prompt:=st.chat_input("Enter your query"):
|
|
57 |
|
58 |
with st.chat_message("assistant"):
|
59 |
st.markdown(response)
|
|
|
|
|
|
|
|
|
60 |
|
61 |
st.session_state.messages.append({
|
62 |
'role':'assistant',
|
|
|
16 |
|
17 |
pipe=pipeline('text2text-generation', model=model, tokenizer=tokenizer)
|
18 |
|
19 |
+
sound_file = BytesIO()
|
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 |
|
|
|
37 |
for gr in greetings:
|
38 |
with st.chat_message("assistant"):
|
39 |
st.markdown(gr)
|
40 |
+
|
41 |
+
tts=gTTS(gr)
|
42 |
+
tts.write_to_fp(sound_file)
|
43 |
+
st.audio(sound_file)
|
44 |
|
45 |
st.session_state.messages.append({
|
46 |
'role':'assistant',
|
|
|
61 |
|
62 |
with st.chat_message("assistant"):
|
63 |
st.markdown(response)
|
64 |
+
|
65 |
+
tts=gTTS(response)
|
66 |
+
tts.write_to_fp(sound_file)
|
67 |
+
st.audio(sound_file)
|
68 |
|
69 |
st.session_state.messages.append({
|
70 |
'role':'assistant',
|