fix

app.py

@@ -18,7 +18,6 @@ from models.models import BiFormer
 
 DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 # DEVICE = torch.device('cpu')
-print(DEVICE)
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -53,7 +52,6 @@ def initialize_components(config_path='config.toml'):
         positional_encoding=False,
         dropout=0.15,
         mode='mean',
-        # device="cuda",
         tr_layer_number=5,
         out_features=256,
         num_classes=7
@@ -314,5 +312,6 @@ def create_demo():
 
 
 if __name__ == "__main__":
+    print('START DEMO')
     demo = create_demo()
     demo.launch()