Spaces:
Sleeping
Sleeping
vitorcalvi
commited on
Commit
•
b185999
1
Parent(s):
efabbbd
pre-launch
Browse files
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),
|