lhkklj

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 from gradio.mix import Parallel
 from fastai.vision.all import load_learner
-from gradio.components import Output
 
 def classify_image_color(img):
     from fastai.vision.all import load_learner
@@ -29,14 +28,14 @@ def postprocess(prediction):
     sorted_pred = sorted(prediction.items(), key=lambda x: x[1], reverse=True)
     return sorted_pred
 
-image = gr.inputs.Image(shape=(256, 256))
 
-output_json = Output(label="JSON Output")
+image = gr.inputs.Image(shape=(256, 256))
+label = gr.outputs.Label()
 
 intf = gr.Interface(
     fn=classify_image, 
     inputs=image, 
-    outputs=output_json, 
+    outputs=label, 
     examples="", 
     title="Lego Brick Classifier", 
     layout="vertical"