Deploy OmniAssistant smolagent (Hugging Face Agents Course Unit 1)

Gradio_UI.py

@@ -24,6 +24,18 @@ from smolagents.agents import ActionStep, MultiStepAgent
 from smolagents.memory import MemoryStep
 from smolagents.utils import _is_package_available
 
+try:
+    import spaces
+except Exception:
+    class _MockSpaces:
+        def GPU(self, *args, **kwargs):
+            if args and callable(args[0]):
+                return args[0]
+            def decorator(f):
+                return f
+            return decorator
+    spaces = _MockSpaces()
+
 
 def pull_messages_from_step(
     step_log: MemoryStep,
@@ -189,6 +201,7 @@ class GradioUI:
             if not os.path.exists(file_upload_folder):
                 os.mkdir(file_upload_folder)
 
+    @spaces.GPU(duration=120)
     def interact_with_agent(self, prompt, messages):
         import gradio as gr