remove img_1.jpg

app.py

@@ -17,7 +17,7 @@ import gradio as gr
 # results.save()
 def total_price(predicted):
     price = 0
-    for name in predicted:
+    for name, confidence in predicted:
         if name == "side dish":
             price += 10
         elif name == "purple rice" or name == "white rice" or name == "brown rice":
@@ -39,8 +39,9 @@ def predict(input_image):
     with torch.no_grad():
         results = model(input_image)
 
-    predicted = results.pandas().xyxy[0]["name"]
-
+    results_pd = results.pandas().xyxy[0]
+    name, confidence = results_pd["name"], results_pd["confidence"]
+    predicted = list(zip(name, confidence))
     output_image = results.render()[0]
     price = total_price(predicted)
 
@@ -53,7 +54,7 @@ with gr.Blocks() as demo:
     """
     <h1 align="center">Group 29 - AI Cafeteria Price Evaluator</h1>
     """)
-    examples = ["./examples/img_1.jpg", "./examples/img_2.jpg", "./examples/img_3.jpg", "./examples/img_4.jpg"]
+    examples = ["./examples/img_1.jpg", "./examples/img_2.jpg", "./examples/img_3.jpg"]
     # gr.Interface(inputs=["image"],outputs=["image"],fn=lambda img:model(img).render()[0]).launch()
     gr.Interface(inputs=["image"], outputs=["image", "text"], fn=predict, examples=examples)