1

app.py

@@ -2,8 +2,12 @@ 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 Zero GPU
+import spaces  # Importing spaces to utilize GPU
 
+import logging
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 # Define the tab structure
 TAB_STRUCTURE = [
@@ -12,11 +16,13 @@ TAB_STRUCTURE = [
     ])
 ]
 
-# Decorate GPU-dependent function with Zero GPU
-@spaces.GPU(duration=120)  # Allocates GPU for 120 seconds when needed
+# Decorate GPU-dependent function with GPU
+@spaces.GPU(duration=300)  # Increased duration if necessary
 def create_demo():
-    
-    # Gradio blocks to create the interface
+    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
     with gr.Blocks(css=CUSTOM_CSS) as demo:
         gr.Markdown(HEADER_HTML)
         with gr.Tabs(elem_classes=["main-tab"]):
@@ -25,9 +31,9 @@ def create_demo():
                     with gr.Tabs():
                         for sub_tab, create_fn in sub_tabs:
                             with gr.Tab(sub_tab):
-                                create_fn()
+                                create_fn(device=device)  # Pass device if needed
         gr.HTML(DISCLAIMER_HTML)
-    
+
     return demo
 
 # Create the demo instance