Update client.py

fixed gr.Image().style to gr.Image(height=512)
triton-inference-server · Jan 18, 2024 · 375a0e3 · 375a0e3
1 parent fe3bb16
commit 375a0e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Conceptual_Guide/Part_6-building_complex_pipelines/gui/client.py b/Conceptual_Guide/Part_6-building_complex_pipelines/gui/client.py
@@ -60,7 +60,7 @@ def generate(prompt):
 with gr.Blocks() as app:
     prompt = gr.Textbox(label="Prompt")
     submit_btn = gr.Button("Generate")
-    img_output = gr.Image().style(height=512)
+    img_output = gr.Image(height=512)
     submit_btn.click(fn=generate, inputs=prompt, outputs=img_output)
 
 app.launch(server_name="0.0.0.0")