Deploy OmniAssistant smolagent (Hugging Face Agents Course Unit 1)

Gradio_UI.py

@@ -272,6 +272,7 @@ class GradioUI:
                 ),
                 resizable=True,
                 scale=1,
+                type="messages",
             )
             # If an upload folder is provided, enable the upload feature
             if self.file_upload_folder is not None:
@@ -289,7 +290,14 @@ class GradioUI:
                 [stored_messages, text_input],
             ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
 
-        demo.launch(debug=True, share=True, **kwargs)
+        is_space = bool(os.getenv("SPACE_ID"))
+        launch_kwargs = {
+            "debug": True,
+            "share": False if is_space else True,
+            "ssr_mode": False,
+        }
+        launch_kwargs.update(kwargs)
+        demo.launch(**launch_kwargs)
 
 
 __all__ = ["stream_to_gradio", "GradioUI"]