Upload app.py

app.py

@@ -13,7 +13,19 @@ import gradio as gr
 # results = model(imgs)  # batched inference
 # results.print()
 # results.save()
+def predict(input_image):
+    """
+    Predict model output
+    """
+    results = model(input_image)
+    output_image = results.render()[0]
+    price = "0"
 
+    # Return the output image and price
+    return [output_image, price]
+    # return [input_image, price]
 
 
-gr.Interface(inputs=["image"],outputs=["image"],fn=lambda img:model(img).render()[0]).launch()
+
+# gr.Interface(inputs=["image"],outputs=["image"],fn=lambda img:model(img).render()[0]).launch()
+gr.Interface(inputs=["image"], outputs=["image", "text"], fn=predict).launch()