Update app.py

app.py

@@ -101,7 +101,8 @@ def process_image(img):
     buffered = io.BytesIO()
     pil_image.save(buffered, format="PNG")
     img_base64 = base64.b64encode(buffered.getvalue()).decode()
-    
+
+    # Return the detected weight and processed image for Gradio
     return f"{weight} kg (Confidence: {confidence:.2f}%)", ist_time, img_base64, gr.update(visible=True)
 
 # Gradio Interface Setup for Hugging Face