Update app.py

app.py

@@ -26,7 +26,7 @@ with gr.Blocks() as demo:
     # Create columns for each model
     with gr.Column(visible=True) as llama_column:
         gr.ChatInterface(
-            fn=lambda message, history: llama_demo.fn(message)["generated_text"],
+            fn=lambda message, history: next(llama_demo.fn(message)),
             title="Llama 3.1 Tulu",
             description="A chat interface for the Llama 3.1 Tulu model"
         )