Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoTokenizer, TFBlenderbotForConditionalGeneration
 import tensorflow as tf
 print("Loading the model......")
 model_name = "WICKED4950/Irisonego5"
@@ -7,7 +7,7 @@ strategy = tf.distribute.MirroredStrategy()
 tf.config.optimizer.set_jit(True)  # Enable XLA
 tokenizer = AutoTokenizer.from_pretrained(model_name) 
 with strategy.scope():
-    model = AutoModelForCausalLM.from_pretrained(model_name)
+    model = TFBlenderbotForConditionalGeneration.from_pretrained(model_name)
 
 print("Interface getting done....")
 # Define the chatbot function