fix server startup, use nvidia runtime

text_to_image.py

@@ -63,6 +63,7 @@ def start_server():
 
     client.run(node=TARGET_NODE,
                image=DOCKER_IMAGE,
+               runtime="nvidia",
                ports=[(5000, 8080)],
                detach=True)