vitorcalvi commited on
Commit
b185999
1 Parent(s): efabbbd

pre-launch

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -15,10 +15,14 @@ from tabs.skin_analysis import create_skin_conductance_tab
15
  from tabs.FACS_analysis_sad import create_facs_analysis_sad_tab
16
  from tabs.roberta_chatbot import create_roberta_chatbot_tab
17
  import spaces
 
18
 
19
  # Import the UI components
20
  from ui_components import CUSTOM_CSS, HEADER_HTML, DISCLAIMER_HTML
21
 
 
 
 
22
  TAB_STRUCTURE = [
23
  ("Visual Analysis", [
24
  ("Emotional Face Expressions", create_face_expressions_tab),
 
15
  from tabs.FACS_analysis_sad import create_facs_analysis_sad_tab
16
  from tabs.roberta_chatbot import create_roberta_chatbot_tab
17
  import spaces
18
+ import torch
19
 
20
  # Import the UI components
21
  from ui_components import CUSTOM_CSS, HEADER_HTML, DISCLAIMER_HTML
22
 
23
+ # Check for GPU availability
24
+ device = "cuda" if torch.cuda.is_available() else "cpu"
25
+
26
  TAB_STRUCTURE = [
27
  ("Visual Analysis", [
28
  ("Emotional Face Expressions", create_face_expressions_tab),