Merge pull request #30 from markstur/fix_gradio

  Fix to work with updated gradio

caikit_huggingface_demo/app.py

@@ -131,7 +131,7 @@ def start_frontend(backend, inference_service):
     channel = grpc.insecure_channel(target)
     frontend = get_frontend(channel, inference_service, module_models)
     print(f"▶️  Starting the frontend gradio UI with using backend target={target}")
-    frontend.launch(share=False, show_tips=False)
+    frontend.launch(share=False)
     print("⏹️  Stopped")
 
 

caikit_huggingface_demo/client/embeddings.py

@@ -33,7 +33,7 @@ def fn(self, model, text_in):
         columns = []
         rows = []
         for k in response.output:
-            columns.append(k.input)
+            columns.append(str(k.input))
 
         for r, _ in enumerate(response.output[0].output):
             row = []

caikit_huggingface_demo/client/image_segmentation.py

@@ -87,8 +87,11 @@ def optional_tab(cls, models, request, predict):
                 with gr.Column():
                     inputs = gr.Image(label="Input Image")
                     output_image = gr.Gallery(
-                        label="Output Masks", show_label=True
-                    ).style(preview=True, object_fit="scale-down")
+                        label="Output Masks",
+                        show_label=True,
+                        preview=True,
+                        object_fit="scale-down",
+                    )
                 with gr.Column():
                     outputs = gr.Label(label="Segments Detected")
                 inputs.change(

requirements.txt

@@ -7,5 +7,5 @@ timm==0.9.1
 sentence-transformers==2.2.2
 transformers[torch]==4.30.0
 # frontend only
-gradio==3.36.1
+gradio==4.11.0
 grpcio==1.53.2