Update app.py

app.py

@@ -350,7 +350,7 @@ messages = [
 def classify(platform, UserInput, Images, Textbox2, Textbox3):
     if Textbox3 == code:
         imageData = None
-        if Image is not None:
+        if Images is not None:
             output = []
             headers = {
                 "Authorization": f"Bearer {auth2}"
@@ -358,14 +358,11 @@ 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}")
 
                     full_path = os.path.join(os.getcwd(), filename)
@@ -376,8 +373,11 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
             else:
                 pass
 
+            image = cv.imread(full_path)
+            image = cv.resize(image, (224, 224))
+            image_array = np.asarray(image)
             image_data = cv.resize(imageData, (224, 224))
-            normalized_image_array = (image_data.astype(np.float32) / 127.0) - 1
+            normalized_image_array = (image_array.astype(np.float32) / 127.0) - 1
             data[0] = normalized_image_array
 
             prediction = model.predict(data)