modifiy app

app.py

@@ -99,7 +99,10 @@ def gen_semantic_feature(frame_dir, semantic_dir):
     sorted_file_names = sorted(file_names)
 
     output_path = semantic_dir / "semantic.npy"
-    features = torch.cuda.FloatTensor(len(sorted_file_names), 768).fill_(0)
+    if torch.cuda.is_available():
+        features = torch.cuda.FloatTensor(len(sorted_file_names), 768).fill_(0)
+    else:
+        features = torch.FloatTensor(len(sorted_file_names), 768).fill_(0)
 
     for idx, file_name in enumerate(sorted_file_names):
         fpath = frame_dir / file_name