Update app.py

app.py

@@ -359,26 +359,19 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
             if platform == "wh":
                 get_image = requests.get(Images, headers=headers)
                 if get_image.status_code == 200:
-                    print(get_image.content)
-                    random_id = random.randint(1000, 9999)
-                    file_extension = ".png"
-                    filename = f"image_{random_id}{file_extension}"
-                    with open(filename, "wb") as file:
-                        file.write(get_image.content)
-                    print(f"Saved image as: {filename}")
+                    image_data = get_image.content
             elif platform == "web":
                 print("WEB")
                 # Handle web case if needed
             else:
                 pass
 
-            image = cv.imread(filename)
+            image = cv.imdecode(np.frombuffer(image_data, np.uint8), cv.IMREAD_COLOR)
             image = cv.resize(image, (224, 224))
             image_array = np.asarray(image)
-            image_data = cv.resize(imageData, (224, 224))
             normalized_image_array = (image_array.astype(np.float32) / 127.0) - 1
             data[0] = normalized_image_array
-
+            
             prediction = model.predict(data)
 
             max_label_index = None