Logeswaransr commited on
Commit
03b857c
1 Parent(s): c885860

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -6,9 +6,12 @@ Created on Thu Sep 21 22:17:43 2023
6
  """
7
 
8
  import streamlit as st
9
- from transformers import pipeline
10
 
11
- pipe=pipeline('sentiment-analysis')
 
 
 
12
 
13
  st.set_page_config(page_title='Sample Chatbot', layout='wide')
14
 
 
6
  """
7
 
8
  import streamlit as st
9
+ from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration
10
 
11
+ model=T5ForConditionalGeneration.from_pretrained("").to("cuda")
12
+ tokenizer=T5Tokenizer.from_pretrained("")
13
+
14
+ pipe=pipeline('text2text-generation', model=model, tokenizer=tokenizer)
15
 
16
  st.set_page_config(page_title='Sample Chatbot', layout='wide')
17