Logeswaransr commited on
Commit
555d0f1
1 Parent(s): 8d63ce3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ Created on Thu Sep 21 22:17:43 2023
8
  import streamlit as st
9
  from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration
10
 
11
- model=T5ForConditionalGeneration.from_pretrained("Logeswaransr/results_T5").to("cuda")
12
  tokenizer=T5Tokenizer.from_pretrained("Logeswaransr/results_T5")
13
 
14
  pipe=pipeline('text2text-generation', model=model, tokenizer=tokenizer)
@@ -18,7 +18,7 @@ st.set_page_config(page_title='Sample Chatbot', layout='wide')
18
  if 'messages' not in st.session_state:
19
  st.session_state.messages=[]
20
 
21
- st.subheader("Sample Chatbot")
22
 
23
  for message in st.session_state.messages:
24
  with st.chat_message(message['role']):
 
8
  import streamlit as st
9
  from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration
10
 
11
+ model=T5ForConditionalGeneration.from_pretrained("Logeswaransr/results_T5").to("cpu")
12
  tokenizer=T5Tokenizer.from_pretrained("Logeswaransr/results_T5")
13
 
14
  pipe=pipeline('text2text-generation', model=model, tokenizer=tokenizer)
 
18
  if 'messages' not in st.session_state:
19
  st.session_state.messages=[]
20
 
21
+ st.subheader("Mine AI")
22
 
23
  for message in st.session_state.messages:
24
  with st.chat_message(message['role']):