Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +44,7 @@ def process_images(videos, x, y):
|
|
44 |
|
45 |
|
46 |
title = "Modelling Human Visual Motion Processing with Trainable Motion Energy Sensing and a Self-attention Network "
|
47 |
-
description = "## Introduction
|
48 |
" The intersection of cognitive neuroscience and computer vision offers exciting advancements in " \
|
49 |
"how machines perceive motion. Our research bridges the gap between these fields by proposing a novel " \
|
50 |
"image-computable model that aligns with human motion perception mechanisms. By integrating trainable" \
|
@@ -88,7 +88,7 @@ if __name__ =='__main__':
|
|
88 |
# # Tesla T4
|
89 |
|
90 |
model = FFV1MT_MS.FFV1DNN()
|
91 |
-
device = "
|
92 |
|
93 |
print('Number fo parameters: {}'.format(model.num_parameters()))
|
94 |
model.to(device)
|
|
|
44 |
|
45 |
|
46 |
title = "Modelling Human Visual Motion Processing with Trainable Motion Energy Sensing and a Self-attention Network "
|
47 |
+
description = "## Introduction 🔥🔥🔥\n" \
|
48 |
" The intersection of cognitive neuroscience and computer vision offers exciting advancements in " \
|
49 |
"how machines perceive motion. Our research bridges the gap between these fields by proposing a novel " \
|
50 |
"image-computable model that aligns with human motion perception mechanisms. By integrating trainable" \
|
|
|
88 |
# # Tesla T4
|
89 |
|
90 |
model = FFV1MT_MS.FFV1DNN()
|
91 |
+
device = "cuda" if torch.cuda.is_available() else "cpu"
|
92 |
|
93 |
print('Number fo parameters: {}'.format(model.num_parameters()))
|
94 |
model.to(device)
|