Update app.py

app.py

@@ -47,8 +47,8 @@ def respond(
 
 # Build Gradio app
 def create_demo():
-    return gr.Blocks().add(
-        gr.Markdown("# AI Model Comparison Tool 🌟"),
+    with gr.Blocks() as demo:
+        gr.Markdown("# AI Model Comparison Tool 🌟")
         gr.ChatInterface(
             respond,
             type="messages",
@@ -72,8 +72,8 @@ def create_demo():
                     value=["Llama"]
                 ),
             ],
-        ),
-    )
+        )
+    return demo
 
 if __name__ == "__main__":
     demo = create_demo()