1

app.py

@@ -2,12 +2,8 @@ import gradio as gr
 import torch
 from tabs.FACS_analysis import create_facs_analysis_tab
 from ui_components import CUSTOM_CSS, HEADER_HTML, DISCLAIMER_HTML
-import spaces  # Importing spaces to utilize GPU if available
+import spaces  # Importing spaces to utilize Zero GPU
 
-import logging
-
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
 
 # Define the tab structure
 TAB_STRUCTURE = [
@@ -16,11 +12,11 @@ TAB_STRUCTURE = [
     ])
 ]
 
+# Decorate GPU-dependent function with Zero GPU
+@spaces.GPU(duration=120)  # Allocates GPU for 120 seconds when needed
 def create_demo():
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-    logger.info(f"Using device: {device}")
-
-    # Ensure that any models loaded within create_facs_analysis_tab use the correct device
+    
+    # Gradio blocks to create the interface
     with gr.Blocks(css=CUSTOM_CSS) as demo:
         gr.Markdown(HEADER_HTML)
         with gr.Tabs(elem_classes=["main-tab"]):
@@ -29,12 +25,12 @@ def create_demo():
                     with gr.Tabs():
                         for sub_tab, create_fn in sub_tabs:
                             with gr.Tab(sub_tab):
-                                create_fn(device=device)  # Pass device if needed
+                                create_fn()
         gr.HTML(DISCLAIMER_HTML)
-
+    
     return demo
 
-# Create the demo instance without GPU decorator
+# Create the demo instance
 demo = create_demo()
 
 if __name__ == "__main__":