Update app.py

app.py

@@ -101,7 +101,7 @@ def create_demo():
 
 if __name__ == "__main__":
     model = Mamba(num_layers = 2, d_input = 1024, d_model = 512, num_classes=7, model_name='jina', pooling=None).to(device)
-    checkpoint = torch.load("Mamba_jina_checkpoint.pth", map_location=torch.device('cpu')
+    checkpoint = torch.load("Mamba_jina_checkpoint.pth", map_location=torch.device('cpu'))
     model.load_state_dict(checkpoint['model_state_dict'])
     demo = create_demo()
     demo.launch()