Update app.py

app.py

@@ -3,7 +3,7 @@
 # !pip install transformers
 # !pip install gradio
 
-from transformers import AutoModelForCausalLM, AutoTokenizer, set_seed, pipeline
+from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer, AutoModelForSeq2SeqLM, set_seed, pipeline
 import gradio as gr
 
 import torch
@@ -11,7 +11,6 @@ torch.set_default_tensor_type(torch.cuda.FloatTensor)
 
 ### need more GPU power to call better models !!!!!!
 
-# from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 # model = AutoModelForSeq2SeqLM.from_pretrained("bigscience/T0pp", use_cache=True) # 11B param
 # tokenizer = AutoTokenizer.from_pretrained("bigscience/T0pp")