Francesco Capuano commited on
Commit
0ea6096
·
1 Parent(s): cb01674

fix: minimal gradio app?

Browse files
lerobot/scripts/server/policy_server.py CHANGED
@@ -327,13 +327,20 @@ class PolicyServer(async_inference_pb2_grpc.AsyncInferenceServicer):
327
 
328
 
329
  def serve():
 
 
 
 
 
 
 
 
330
  server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
331
  async_inference_pb2_grpc.add_AsyncInferenceServicer_to_server(PolicyServer(), server)
332
  server.add_insecure_port("[::]:50051")
333
  server.start()
334
  logger.info("PolicyServer started on port 50051")
335
 
336
- print("Hello world :) Server started!")
337
 
338
  try:
339
  while True:
 
327
 
328
 
329
  def serve():
330
+ import gradio as gr
331
+
332
+ def greet(name):
333
+ return "Hello " + name + "!"
334
+
335
+ demo = gr.Interface(fn=greet, inputs="text", outputs="text")
336
+ demo.launch()
337
+
338
  server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
339
  async_inference_pb2_grpc.add_AsyncInferenceServicer_to_server(PolicyServer(), server)
340
  server.add_insecure_port("[::]:50051")
341
  server.start()
342
  logger.info("PolicyServer started on port 50051")
343
 
 
344
 
345
  try:
346
  while True:
pyproject.toml CHANGED
@@ -74,6 +74,7 @@ dependencies = [
74
  "wandb>=0.16.3",
75
  "zarr>=2.17.0",
76
  "grpcio>=1.71.0",
 
77
  ]
78
 
79
  [project.optional-dependencies]
 
74
  "wandb>=0.16.3",
75
  "zarr>=2.17.0",
76
  "grpcio>=1.71.0",
77
+ "gradio"
78
  ]
79
 
80
  [project.optional-dependencies]