Update app.py

app.py

@@ -133,13 +133,18 @@ if selected_type == "Upload Image":
                 img_bytes = image.read()
 
             img = Image.open(io.BytesIO(img_bytes))
-    
-            #for img in results.imgs:
-             #   img_base64 = Image.fromarray(img)
-              #  img_base64.save(downloaded_image, format="JPEG")
 
-            img = Image.fromarray(img)
-            results = FINAL(img)    
+
+            img = np.asarray(bytearray(img.read()), dtype=np.uint8)
+            
+            img = cv2.imdecode(img, cv2.IMREAD_COLOR)
+
+            results = FINAL(img)  
+    
+            for img in results.imgs:
+                img_base64 = Image.fromarray(img)
+                img_base64.save(downloaded_image, format="JPEG")
+              
 
             final_image = Image.open(downloaded_image)
             print("Opening ",final_image)