try restricting batch size to 2 to avoid excessive memory use

chanind · May 19, 2022 · 4f2e58c · 4f2e58c
1 parent cfc4d21
commit 4f2e58c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo/server/app.py b/demo/server/app.py
@@ -10,7 +10,7 @@
 CORS(app)
 
 
-transformer = FrameSemanticTransformer()
+transformer = FrameSemanticTransformer(max_batch_size=2)
 
 
 @app.errorhandler(400)