Spaces:

hertogateis
/

SmallBot

Sleeping

hertogateis commited on Jan 9

Commit

7db7138

verified ·

1 Parent(s): 7871f39

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import streamlit as st
 import transformers
 from transformers import T5Tokenizer, T5ForConditionalGeneration
-model_name = "t5-large"
 tokenizer = T5Tokenizer.from_pretrained(model_name)
 model = T5ForConditionalGeneration.from_pretrained(model_name)
 def generate_response(input_text):

 import streamlit as st
 import transformers
 from transformers import T5Tokenizer, T5ForConditionalGeneration
+model_name = "t5-base"
 tokenizer = T5Tokenizer.from_pretrained(model_name)
 model = T5ForConditionalGeneration.from_pretrained(model_name)
 def generate_response(input_text):