Spaces:

crisrm128
/

YOLOS-tiny-Docker

Sleeping

crisrm128 commited on Nov 22, 2023

Commit

a552163

1 Parent(s): d63daaa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -45,18 +45,8 @@ def detect_objects(url: str):
         image_byte_array = io.BytesIO()
         image.save(image_byte_array, format="PNG")
-        # Encode the byte stream as a base64 string
-        image_base64 = base64.b64encode(image_byte_array.getvalue()).decode("utf-8")
-        # Create a custom HTML response
-        html_content = f"""
-        <html>
-        <body>
-            <img src="data:image/png;base64,{image_base64}" alt="Detected Objects">
-        </body>
-        </html>
-        """
-        return HTMLResponse(content=html_content)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error processing image: {str(e)}")

         image_byte_array = io.BytesIO()
         image.save(image_byte_array, format="PNG")
+        # Return the image as a Response with content type "image/png"
+        return Response(content=image_byte_array.getvalue(), media_type="image/png")
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error processing image: {str(e)}")