Logeswaransr commited on
Commit
48bf45d
·
1 Parent(s): 6900593

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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')