sunana commited on
Commit
b5ea36b
•
1 Parent(s): e5d21c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(^_^)\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,7 +88,7 @@ if __name__ =='__main__':
88
  # # Tesla T4
89
 
90
  model = FFV1MT_MS.FFV1DNN()
91
- device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"
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)