Update app.py

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
-# xl size run out of memory on 16GB vm
-tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-large")
-model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large")
+# xl size run out of memory on 16GB VM
+# google/flan-t5-large
+model_name = 'jncraton/fastchat-t5-3b-v1.0-ct2-int8'
+
+tokenizer = T5Tokenizer.from_pretrained(model_name)
+model = T5ForConditionalGeneration.from_pretrained(model_name)
 
 title = ""