Update app.py

app.py

@@ -2,10 +2,7 @@ import huggingface_hub as hf_hub
 import gradio as gr 
 import os
 
-client = hf_hub.InferenceClient(
-    model = 'google/flan-t5-xxl',
-    token = os.environ['HUGGINGFACEHUB_API_TOKEN']
-)
+client = hf_hub.InferenceClient(token = os.environ['HUGGINGFACEHUB_API_TOKEN'])
 
 def create_prompt(question, history):
     template = '''You are a good chatbot that talks with human. 
@@ -31,7 +28,7 @@ def create_prompt(question, history):
 
 def chat_interface(question, history):
     prompt = create_prompt(question, history)
-    response = client.text_generation(prompt)
+    response = client.text_generation(prompt, model = 'google/flan-t5-xxl')
     return response
 
 chat_app = gr.ChatInterface(fn = chat_interface)